1563-石子游戏Ⅴ-区间DP问题

题目

分析

题意还是比较好理解的,每次将石子分成两大堆,抛弃总和大的那一堆,留下少的一堆并且总分数中加上少的一堆的和,直至只剩下一个石头,游戏结束。而题目给的数据规模在500,那么算法的复杂度要在O(n²)以内。
首先考虑一下暴力解法。当石子大于一块时,遍历每一个可能的分割位置,分别计算两边的和,留下少的一堆并在答案中加上少的那堆的和,比较所有的可能,找到最大的分数,其中计算部分和可以使用前缀和数组预处理来减少时间复杂度。而在暴力解法中,会出现很多区间重复计算的问题,动态规划显然是一个好的处理方式。

区间动态规划

状态定义

由于是区间上的动态规划问题,定义时,可以使用二维数组dp[i][j],枚举所有区间,也就是 i 表示区间头,j 表示区间尾,根据题意,可以想到dp数组的定义:dp[i][j]表示在区间(i, j)上,Alice能获得的最大分数,由此可以得到基础状态:

  • 对于长度为一的区间,只剩一块石子,游戏结束,即dp[i][i] = 0
状态转移

由于要找到所有的分割方法中的最优方法,所以状态转移时肯定需要遍历每个分割位置取最大值。然后对于每个分割位置,根据题意,取和小的那一堆,然后对小的那堆继续分割,小的那一堆也就相当于sum[j] - sum[i],而小的那堆继续分割的值在之前已经计算过,为dp[i][j],所以可以得到:

    if (sum[k + 1] - sum[i] > sum[j + 1] - sum[k + 1]) {
        dp[i][j] = Math.max(dp[i][j], sum[j + 1] - sum[k + 1] + dp[k + 1][j]);
    } else if (sum[k + 1] - sum[i] < sum[j + 1] - sum[k + 1]) {
        dp[i][j] = Math.max(dp[i][j], sum[k + 1] - sum[i] + dp[i][k]);
    }

这里的 i 表示区间头,j 表示区间尾,k 表示分割位置,sum 表示前缀和数组,下标要慎重考虑。
除了这样两种情况,还有两堆相等的情况,两堆相等并不能给出明确的选择给出哪个更好,所以应该计算左堆和右堆,取其中的最大值:

    dp[i][j] = Math.max(dp[i][j],
               Math.max(sum[j + 1] - sum[k + 1] + dp[k + 1][j], sum[k + 1] - sum[i] + dp[i][k]));

这里每次取dp[i][j]和另一个值比较是为了遍历分割位置 k 时取到所以结果中的最大值。

完整代码

有了状态定义、初始状态及状态转移,就可以写出DP代码了,通常情况的区间DP问题采用先循环遍历区间长度,再循环遍历区间头来遍历所有可能并且保证后边用到的值在先前已经得到。

class Solution {
    private int[][] dp;

    public int stoneGameV(int[] a) {
        int n = a.length;
        // 前缀和
        int[] sum = new int[a.length + 1];
        for (int i = 0; i < a.length; i++) {
            sum[i + 1] = sum[i] + a[i];
        }

        dp = new int[n][n];

        for (int len = 2; len <= n; len++) {
            for (int i = 0; i + len - 1 < n; i++) {
                int j = i + len - 1;
                for (int k = i; k < j; k++) {
                    if (sum[k + 1] - sum[i] > sum[j + 1] - sum[k + 1]) {
                        dp[i][j] = Math.max(dp[i][j], sum[j + 1] - sum[k + 1] + dp[k + 1][j]);
                    } else if (sum[k + 1] - sum[i] < sum[j + 1] - sum[k + 1]) {
                        dp[i][j] = Math.max(dp[i][j], sum[k + 1] - sum[i] + dp[i][k]);
                    } else {
                        dp[i][j] = Math.max(dp[i][j],
                                Math.max(sum[j + 1] - sum[k + 1] + dp[k + 1][j], sum[k + 1] - sum[i] + dp[i][k]));
                    }
                }
            }
        }
        return dp[0][n - 1];
    }
}

用时在500ms左右,也可以采用记忆化递归来处理,时间可以达到70ms左右,这里我也不太懂原因是什么,理论上循环计算要比递归调用用时短一些,可力扣上好多这样的题都是记忆化递归快,不是很懂他运行的机制,改写成递归还是比较简单的,只要把k部分的循环写到递归里即可,我这里就不在给出代码了。

总结

DP问题可真是老大难,各种各样的形式,思想差不多但是方法不同的解法,dp数组的定义,没有一样很容易掌握,但还是需要去学习啊。区间DP也是一个比较常见的DP问题类型,一点点努力,一点点进步~
文章如果有不正确的地方还请指出,感恩相遇~~

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 160,881评论 4 368
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 68,052评论 1 301
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 110,598评论 0 250
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,407评论 0 217
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,823评论 3 294
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,872评论 1 224
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 32,037评论 2 317
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,778评论 0 204
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,505评论 1 247
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,745评论 2 253
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,233评论 1 264
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,568评论 3 260
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,231评论 3 241
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,141评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,939评论 0 201
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,954评论 2 283
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,784评论 2 275