395. 硬币排成线 II
有 n
个不同价值的硬币排成一条线。两个参赛者轮流从左边依次拿走 1 或 2 个硬币,直到没有硬币为止。计算两个人分别拿到的硬币总价值,价值高的人获胜。
请判定 第一个玩家 是输还是赢?
样例
给定数组 A = [1,2,2]
, 返回 true
.
给定数组 A = [1,2,4]
, 返回 false
.
思路:动态规划,状态转移方程为:DP[i] = max{nums[i]+min{DP[i+2],DP[i+3]} , nums[i]+nums[i+1]+min{DP[i+3],DP[i+4]} }
详见:https://www.cnblogs.com/Revenent-Blog/p/7587766.html
1 class Solution { 2 public: 3 /** 4 * @param values: a vector of integers 5 * @return: a boolean which equals to true if the first player will win 6 */ 7 bool firstWillWin(vector<int> &values) { 8 if(values.size()<=2)return true; 9 int n=values.size(); 10 vector<int> dp(n, 0); 11 dp[n-1]=values[n-1]; 12 dp[n-2]=values[n-1]+values[n-2]; 13 for(int i=n-3; i>=0; --i){ 14 int a3=0; 15 int a4=0; 16 if(i+3<n)a3=dp[i+3]; 17 if(i+4<n)a4=dp[i+4]; 18 dp[i]=max(values[i]+min(dp[i+2], a3), values[i]+values[i+1]+min(a3, a4)); 19 } 20 int sum=0; 21 for(int i=0; i<n; ++i)sum=sum+values[i]; 22 if(dp[0]>sum-dp[0])return true; 23 else return false; 24 } 25 };