leetcode 1937. 扣分后的最大得分「动态规划」「拆项」

1937. 扣分后的最大得分

题目描述:

给你一个n*m的整数矩阵ar,一开始你的得分为0,你想最大化从矩阵中得到的分数

你的得分方式为:每一行 中选取一个格子,选中坐标为 (r, c) 的格子会给你的总得分 增加 points[r][c]

然而,相邻行之间被选中的格子如果隔得太远,你会失去一些得分。对于相邻行 rr + 1 (其中 0 <= r < m - 1),选中坐标为 (r, c1)(r + 1, c2) 的格子,你的总得分 减少 abs(c1 - c2)

请你返回你能得到的 最大 得分

思路:

很经典的拆项思路

d p [ i ] [ j ] dp[i][j] dp[i][j]表示前 i − 1 i-1 i−1行均选择了一个格子,第 i i i行选了第 j j j个格子,获得的最大价值

很显然状态要从上一行转移过来

  • d p [ i ] [ j ] = m a x ( d p [ i ] [ j ] , a r [ i ] [ j ] + d p [ i − 1 ] [ k ] − a b s ( k − j ) ) dp[i][j] = max(dp[i][j], ar[i][j] + dp[i - 1][k] - abs(k - j)) dp[i][j]=max(dp[i][j],ar[i][j]+dp[i−1][k]−abs(k−j))

如果这样暴力转移,则时间复杂度是 O ( n ∗ m ∗ m ) O(n*m*m) O(n∗m∗m),一定会超时的

我们考虑拆项,根据 k k k和 j j j的大小,分情况讨论

  • d p [ i ] [ j ] = m a x ( d p [ i ] [ j ] , t r [ i ] [ j ] + ( d p [ i − 1 ] [ k ] + k ) − j ) , k < = j dp[i][j] = max(dp[i][j], tr[i][j] + (dp[i-1][k] + k) - j), k <= j dp[i][j]=max(dp[i][j],tr[i][j]+(dp[i−1][k]+k)−j),k<=j
  • d p [ i ] [ j ] = m a x ( d p [ i ] [ j ] , t r [ i ] [ j ] + ( d p [ i − 1 ] [ k ] − k ) + j ) , k > j dp[i][j] = max(dp[i][j], tr[i][j] + (dp[i-1][k] - k) + j), k > j dp[i][j]=max(dp[i][j],tr[i][j]+(dp[i−1][k]−k)+j),k>j

对于 d p [ i ] [ j ] dp[i][j] dp[i][j]我们如果可以用 O ( 1 ) O(1) O(1)的时间得到 m a x { d p [ i − 1 ] [ k ] + k } , k < = j max\{dp[i-1][k] + k\},k<=j max{dp[i−1][k]+k},k<=j 和 m a x { d p [ i − 1 ] [ k ] − k } max\{dp[i-1][k]-k\} max{dp[i−1][k]−k}的值就行了

所以我们记录一个前缀最大值数组和一个后缀最大值数组,分别记录即可

同时,我们可以发现转移的时候只和上一维度的值有关,可以用滚动数组优化空间

所以最后的时间复杂度是 O ( n ∗ m ) O(n*m) O(n∗m)

cpp 复制代码
class Solution {
public:
    long long maxPoints(vector<vector<int>>& tr) {
        int n = tr.size(), m = tr[0].size();
        vector<long long>dp(m + 1);
        vector<long long>pre(m + 2), suf(m + 2);
        for(int j = 1; j <= m + 1; ++j){
            pre[j] = j;
            suf[j] = -j;
        }
        for(int i = 1; i <= n; ++i){   
            for(int j = 1; j <= m; ++j){
                dp[j] = tr[i - 1][j - 1] + max(pre[j] - j, suf[j + 1] + j);
            }
            for(int j = 1; j <= m; ++j){
                pre[j] = max(pre[j - 1], dp[j] + j);
            }
            for(int j = m; j >= 1; --j){
                suf[j] = max(suf[j + 1], dp[j] - j);
            }
        }
        long long ans = 0;
        for(int i = 1; i <= m; ++i)ans = max(ans, dp[i]);
        return ans;
    }
};
相关推荐
爱爬山的老虎6 分钟前
【面试经典150题】LeetCode121·买卖股票最佳时机
数据结构·算法·leetcode·面试·职场和发展
雾月5532 分钟前
LeetCode 914 卡牌分组
java·开发语言·算法·leetcode·职场和发展
想跑步的小弱鸡35 分钟前
Leetcode hot 100(day 4)
算法·leetcode·职场和发展
Fantasydg36 分钟前
DAY 35 leetcode 202--哈希表.快乐数
算法·leetcode·散列表
jyyyx的算法博客37 分钟前
Leetcode 2337 -- 双指针 | 脑筋急转弯
算法·leetcode
ゞ 正在缓冲99%…1 小时前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong1 小时前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
惊鸿.Jh1 小时前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
好好学习^按时吃饭2 小时前
[NOIP 1999 提高组] 导弹拦截
动态规划
想跑步的小弱鸡8 小时前
Leetcode hot 100(day 3)
算法·leetcode·职场和发展