[程序员代码面试指南]递归和动态规划-排成一条线的纸牌博弈问题(DP)
题目
给定一个整型数组arr,代表数值不同的纸牌排成一条线。玩家A和玩家B依次拿走每张纸牌,规定玩家A先拿,玩家B后拿,但是每个玩家每次只能拿走最左或最右的纸牌,玩家A和玩家B都绝顶聪明。请返回最后获胜者的分数。
例子
arr=[1,2,100,4]。
开始时玩家A只能拿走1或4。如果玩家A拿走1,则排列变为[2,100,4],接下来玩家B可以拿走2或4,然后继续轮到玩家A。如果开始时玩家A拿走4,则排列变为[1,2,100],接下来玩家B可以拿走1或100,然后继续轮到玩家A。玩家A作为绝顶聪明的人不会先拿4,因为拿了4之后玩家B将拿走100。所以玩家A会先拿1,让排列变为[2,100,4],接下来玩家B不管怎么选,100都会被玩家A拿走。玩家A会获胜,分数为101。所以返回101。
arr=[1,100,2]。
开始时玩家A不管拿1还是2,玩家B作为绝顶聪明的人,都会把100拿走。玩家B会获胜,分数为100。所以返回100。
题解
- 两个dp数组 offensivePos[i,j]、 defensivePos[i,j] 分别表示对于arr[i,j],当前为先手能够获得的最大得分、当前为后手能够获得的最大的分。
- 初始化:当i=j ,offensivePos[j][j]=arr[j] , defensivePos[i,j] =0.
- 其他位置两个数组的转移方程见代码。
- 遍历顺序是"从上往下从后往左"
- 时间复杂度O(n^2) ,额外空间复杂度O(n^2).
代码
public class Main {
public static void main(String args[]) {
int[] arr= {1,2,100,4};
int winNum=getWinNum(arr);
System.out.println(winNum);
}
public static int getWinNum(int[] arr) {
if(arr.length==0) {
return 0;
}
int[][] offensivePos=new int[arr.length][arr.length];
int[][] defensivePos=new int[arr.length][arr.length];
for(int j=0;j<arr.length;++j) {//由上至下
offensivePos[j][j]=arr[j];
for(int i=j-1;i>=0;--i) {//由右至左
offensivePos[i][j]=Math.max(arr[i]+defensivePos[i+1][j], arr[j]+defensivePos[i][j-1]);//
defensivePos[i][j]=Math.min(offensivePos[i+1][j], offensivePos[i][j-1]);
}
}
return Math.max(defensivePos[0][arr.length-1], offensivePos[0][arr.length-1]);
}
}
posted on 2019-05-30 23:59 coding_gaga 阅读(572) 评论(0) 编辑 收藏 举报