LeetCode - 486. Predict the Winner(排成一条线的纸牌博弈问题)

33 篇文章 0 订阅

LeetCode - 486. Predict the Winner(排成一条线的纸牌博弈问题)

  • 递归解法
  • 动态规划解法

题目链接
题目


递归解法

  • 定义递归函数f[i,j],表示的是如果arr[i...j]这个排列上的纸牌被绝顶聪明的人先拿,最终可以获得什么分数。
  • 定义递归函数s[i,j],表示的是如果arr[i...j]这个排列上的纸牌被绝顶聪明的人后拿,最终可以获得什么分数。

首先对于f[i,j]:

  • 如果i == j,即arr[i...j]上只有一张纸牌,当然会被先拿纸牌的人拿走,所以可以返回arr[i];
  • 如果i != j,此时先拿的人有两种选择。如果拿走arr[i],那么会剩下arr[i+1,j],这个时候,对于当前的玩家来说,他就成了后拿的人,所以他之后能获得的分数为s(i+1,j); 如果他拿走的是arr[j],那么会剩下arr[i...j-1],对于当前的玩家来说,它成了后拿的人,之后能获得的分数为s(i,j-1),因为当前的玩家会做出最好的选择,所以是max(arr[i]+s[i+1,j], arr[j]+s[i,j-1]);

然后来考虑s[i,j] :

  • 如果i == j,即arr[i...j]上只有一张纸牌,作为后拿的人必然什么也得不到,所以返回0
  • 如果i != j,根据s函数的定义,当前的玩家的对手会先拿纸牌,如果对手拿走了arr[i...j]中的arr[i],剩下arr[i+1,j],然后轮到当前玩家拿。如果对手拿走了arr[i...j]中的arr[j],剩下arr[i,j-1],然后轮到当前玩家拿。因为对手会拿走最好的,所以当前玩家只能拿最差的,所以是min(f[i+1,j],f[i,j-1])

所以可以写出如下的代码

class Solution {

    public boolean PredictTheWinner(int[] arr) {
        if (arr == null || arr.length == 0)
            return true;
        return f(arr, 0, arr.length - 1) >= s(arr, 0, arr.length - 1);
    }

    public int f(int[] arr, int i, int j) { //先拿的
        if (i == j)
            return arr[i];
        return Math.max(arr[i] + s(arr, i + 1, j), arr[j] + s(arr, i, j - 1)); //拿了其中一个之后,当前玩家成了后拿的那个人
    }

    public int s(int[] arr, int i, int j) { //后拿的
        if (i == j) //后拿的这个 已经不能拿了
            return 0;
        return Math.min(f(arr, i + 1, j), f(arr, i, j - 1));
    }
}

动态规划解法

上面的方法的时间复杂度达到O(2^n)次方,我们可以根据上面的方法来改出动态规划。

  • 先生成两张二维dp表,f表和s表,我们发现,根据上面的方法,f表中的主对角线上的值就是arr[i]的值(即f[i][i] = arr[i]);
  • 同理s表中的主对角线上的值为0
  • 然后我们可以得到f表中的某个位置f[i,j]依赖的是s表中的位置s[i+1,j]s[i,j-1](即下面的位置和左边的位置);
  • s表中同理依赖f表中f[i+1][j]f[i,j-1]的位置;

具体过程看下图
这里写图片描述

最后我们要求的就是f[0][len]s[0][len]这两个值,如果f[0][len]更大的话,第一个玩家就能赢;

可以写出如下代码

class Solution {

    public boolean PredictTheWinner(int[] nums) {
        if (nums == null || nums.length == 0 || nums.length == 1)
            return true;
        int len = nums.length - 1;
        int[][] f = new int[len + 1][len + 1];
        int[][] s = new int[len + 1][len + 1];
        for (int i = 0; i <= len; i++) {//对角线上填充(就是递归中的边界条件)
            f[i][i] = nums[i];
//            s[i][i] = 0;  // 数组自动初始化
        }
        for (int i = len - 1; i >= 0; i--) {
            for (int j = i + 1; j <= len; j++) {
                f[i][j] = Math.max(nums[i] + s[i + 1][j], nums[j] + s[i][j - 1]); //注意这里是arr[j]+是s[i][j-1]
                s[i][j] = Math.min(f[i + 1][j], f[i][j - 1]);
            }
        }
        return f[0][len] >= s[0][len];
    }
}
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值