leecode.1406.石子游戏3

题目

Alice 和 Bob 用几堆石子在做游戏。几堆石子排成一行，每堆石子都对应一个得分，由数组 stoneValue 给出。

Alice 和 Bob 轮流取石子，Alice 总是先开始。在每个玩家的回合中，该玩家可以拿走剩下石子中的的前 1、2 或 3 堆石子 。比赛一直持续到所有石头都被拿走。

每个玩家的最终得分为他所拿到的每堆石子的对应得分之和。每个玩家的初始分数都是 0 。比赛的目标是决出最高分，得分最高的选手将会赢得比赛，比赛也可能会出现平局。

假设 Alice 和 Bob 都采取 最优策略 。如果 Alice 赢了就返回 “Alice” ，Bob 赢了就返回 “Bob”，平局（分数相同）返回 “Tie” 。

思路分析

• 使用一个长度为n+1的dp数组，其中dp[n] = 0.
• dp[i]表示当前回合中玩家A比玩家B在[i,i+1,…n-1]数组中有的最多的分数。
• 对于dp[i]来说，玩家可以从i位置起，取1，2，3个石头。取第一个石头，分数就变成了stone[i] - dp[i + 1];取前两个，分数变成了stone[i] + stone[i + 1] - dp[i + 2];取前三个，分数就变成了stone[i]+stone[i+1]+stone[i+2] - dp[i+3]。取三个数值里面的最大值。

代码

class Solution {
public:
    string stoneGameIII(vector<int>& stoneValue) {
        int n = stoneValue.size(), sum = 0;
        vector<int> dp(n + 1, -0x3f3f3f3f);
        dp[n] = 0;
        for(int i = n - 1;i >= 0;i--){
            int sum = 0;
            for(int j = 0; j < 3 && i + j < n;j++){
                sum += stoneValue[i + j];
                dp[i] = max(dp[i], sum - dp[i + j + 1]);
            }
        }
        if(dp[0] == 0) return "Tie";
        else if(dp[0] > 0) return "Alice";
        else return "Bob";
    }
};

例子

例如对于[1,2,3,4,5,-9,7]来说，过程中的dp和pos数组的变化情况。上面的数组是dp在此次循环中的数值。下面pos是表示应该从当前下标取几个石头。
######################
[-1061109567 -1061109567 -1061109567 -1061109567 -1061109567 -1061109567 7 0 ]
[0 0 0 0 0 0 1 0 ]
######################
######################
[-1061109567 -1061109567 -1061109567 -1061109567 -1061109567 -2 7 0 ]
[0 0 0 0 0 2 1 0 ]
######################
######################
[-1061109567 -1061109567 -1061109567 -1061109567 7 -2 7 0 ]
[0 0 0 0 1 2 1 0 ]
######################
######################
[-1061109567 -1061109567 -1061109567 11 7 -2 7 0 ]
[0 0 0 2 1 2 1 0 ]
######################
######################
[-1061109567 -1061109567 14 11 7 -2 7 0 ]
[0 0 3 2 1 2 1 0 ]
######################
######################
[-1061109567 2 14 11 7 -2 7 0 ]
[0 3 3 2 1 2 1 0 ]
######################
######################
[-1 2 14 11 7 -2 7 0 ]
[1 3 3 2 1 2 1 0 ]
######################