LeetCode·每日一题·1140.石子游戏 ||·记忆化搜索

作者：小迅

链接：https://leetcode.cn/problems/stone-game-ii/solutions/2126012/ji-yi-hua-sou-suo-zhu-shi-chao-ji-xiang-wvp26/

来源：力扣（LeetCode）

著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

题目

示例

思考

题意 -> 给定一堆数组，两个人从中取数组元素，返回第一个人能取得的最大数组和

可以使用递归枚举所有选择石头的方法，如图：

根据上图，定义 dfs(i,M) 表示从 piles[i] 开始拿石子，可以得到的最大石子数。M 的含义同题目描述。

对于每个节点，由于剩余的石子总数是固定的，如果拿了某几堆石子后，对手能得到的石子数最少，那么自己能得到的石子数就是最多的。

注意当下一个人进行选择时，我们应该让他选择最少的元素，这样才能让我们选择更多的元素，当然但他进行选择时也会让我们选择最少的元素，这样才能做到 爱丽丝和鲍勃都发挥出最佳水平

当然我们在递归过程中会出现很多重复的递归过程，使用记忆化搜索进行优化，举个例子，「Alice 拿一堆，Bob 拿两堆」和「Alice 拿两堆，Bob 拿一堆」，都会递归到 dfs(3,2)。

代码注释超级详细

代码

int dfs(int i, int m, int n, int *s, int (*cache)[(n + 1) / 4 + 1]) {
    if (i + m * 2 >= n) return s[i];//最优解
    int *res = &cache[i][m];
    if ((*res) != -1) return *res;//这种类型之前已经枚举过
    int mn = INT_MAX;
    for (int x = 1; x <= m * 2; ++x)//枚举对手选择
        mn = fmin(mn, dfs(i + x, fmax(m, x), n, s, cache));//取最小值
    return (*res) = s[i] - mn;//返回后缀和之差
};

int stoneGameII(int* piles, int pilesSize){
    for (int i = pilesSize - 2; i >= 0; --i)
        piles[i] += piles[i + 1]; // 后缀和
    int cache[pilesSize][(pilesSize + 1) / 4 + 1];
    memset(cache, -1, sizeof(cache)); // -1 表示没有访问过
    return dfs(0, 1, pilesSize, piles, cache);
}

作者：小迅
链接：https://leetcode.cn/problems/stone-game-ii/solutions/2126012/ji-yi-hua-sou-suo-zhu-shi-chao-ji-xiang-wvp26/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。