2020_9_1 每日一题 486. 预测赢家

给定一个表示分数的非负整数数组。 玩家 1 从数组任意一端拿取一个分数,随后玩家 2 继续从剩余数组任意一端拿取分数,然后玩家 1 拿,……
。每次一个玩家只能拿取一个分数,分数被拿取之后不再可取。直到没有剩余分数可取时游戏结束。最终获得分数总和最多的玩家获胜。

给定一个表示分数的数组,预测玩家1是否会成为赢家。你可以假设每个玩家的玩法都会使他的分数最大化。

示例 1:

输入:[1, 5, 2] 输出:False 解释:一开始,玩家1可以从1和2中进行选择。 如果他选择 2(或者 1 ),那么玩家 2 可以从
1(或者 2 )和 5 中进行选择。如果玩家 2 选择了 5 ,那么玩家 1 则只剩下 1(或者 2 )可选。 所以,玩家 1 的最终分数为
1 + 2 = 3,而玩家 2 为 5 。 因此,玩家 1 永远不会成为赢家,返回 False 。 示例 2:

输入:[1, 5, 233, 7] 输出:True 解释:玩家 1 一开始选择 1 。然后玩家 2 必须从 5 和 7 中进行选择。无论玩家
2 选择了哪个,玩家 1 都可以选择 233 。
最终,玩家 1(234 分)比玩家 2(12 分)获得更多的分数,所以返回 True,表示玩家 1 可以成为赢家。

提示:

1 <= 给定的数组长度 <= 20. 数组里所有分数都为非负数且不会大于 10000000 。 如果最终两个玩家的分数相等,那么玩家 1
仍为赢家。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/predict-the-winner
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

经典DP,由小问题的答案计算大问题的答案。目标是知道先手能获得的最高分,那么需要分别知道去掉头尾之后,先手取得的最高分。已经算过的答案存在哈希表中,还是数据有限时的数组代替哈希表的tricks。另外需要注意一下总分为奇偶时的胜利判据。随手写一个递归的:

class Solution {
    
    int[] result = new int[400];
    
    int maxScore(int[] nums, int l, int r, int leftScore) {
        if(result[l * 20 + r] != -1)
            return result[l * 20 + r];

        int curResult = 0;
        if(l > r) {
            curResult = 0;
        } else if(l == r) {
            curResult = nums[l];
        } else {
            int left = maxScore(nums, l + 1, r, leftScore - nums[l]);
            int right = maxScore(nums, l, r - 1, leftScore - nums[r]);
            curResult = leftScore - Math.min(left, right);
        }
        result[l * 20 + r] = curResult;
        return curResult;
    }

    public boolean PredictTheWinner(int[] nums) {
        Arrays.fill(result, -1);        
        int sum = 0;
        for(int i = 0; i < nums.length; ++i) {
            sum += nums[i];
        }

        return maxScore(nums, 0, nums.length - 1, sum) >= (sum + 1) / 2; 
    }
}

然后日常翻翻题解,en题解的dp用的迭代。不过有趣的是,题解的中间结果算的不是某种情况下的最高分,而是某种情况下的最高分差。这样状态转移方程会更简单一些,也不用每次传递剩余总分了。不过这里需要注意,最高分差会出现负数,不能用特殊值来标记走过没有了。

class Solution {
    
    int[] result = new int[400];
    int[] visited;
    
    int maxScore(int[] nums, int l, int r) {
        if(visited[l * 20 + r] != 0)
            return result[l * 20 + r];

        int curResult;
        if(l > r) {
            curResult = 0;
        } else if(l == r) {
            curResult = nums[l];
        } else {
            int left = nums[l] - maxScore(nums, l + 1, r);
            int right = nums[r] - maxScore(nums, l, r - 1);
            curResult = Math.max(left, right);
        }
        result[l * 20 + r] = curResult;
        visited[l * 20 + r] = 1;
        return curResult;
    }

    public boolean PredictTheWinner(int[] nums) {
        visited = new int[400];       
        return maxScore(nums, 0, nums.length - 1) >= 0; 
    }
}

再仔细看看答案,这个题的结果是一个右上三角(左索引为行右索引为列)。并且先填充对角线上的数值(一个数的情况),然后从下往上逐行计算(计算MaxDiff[i][j] 用到MaxDiff[i - 1][j] 和MaxDiff[i][j - 1],分别是左一格和上一格的数)。这样就可以用一个一行代替多行结果的tricks,好吧我对迭代的递归还是不太熟练,实现了一下:

class Solution {


    public boolean PredictTheWinner(int[] nums) {
        int[] MaxDiff = new int[nums.length];
        for(int i = 0; i < nums.length; ++i) {
            MaxDiff[i] = nums[i];
        }

        for(int i = nums.length - 2; i >= 0; --i)
        for(int j = i + 1; j < nums.length; ++j) {
            MaxDiff[j] = Math.max(nums[i] - MaxDiff[j], nums[j] - MaxDiff[j - 1]);//前一个其实是[i + 1][j]
        }

        return MaxDiff[nums.length - 1] >= 0;
    }
}

嗯,短多了,就是费脑子TAT。

说起来,三角形的DP答案似乎很熟悉啊,算法导论里头刚讲DP的时候好像就是个三角形的例子,去翻了一下,和最优矩阵乘法虽然答案都长得是三角形,但是最有矩阵乘法不管是状态转移还是结果生成都要复杂多了…神书不愧是神书

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值