码迷,mamicode.com
首页 > 其他好文 > 详细

LintCode-硬币排成线 II

时间:2015-05-23 12:57:18      阅读:169      评论:0      收藏:0      [点我收藏+]

标签:面试   lintcode   

有 n 个不同价值的硬币排成一条线。两个参赛者轮流从右边依次拿走 1 或 2 个硬币,直到没有硬币为止。计算两个人分别拿到的硬币总价值,价值高的人获胜。

请判定 第一个玩家 是输还是赢?

样例

给定数组 A = [1,2,2], 返回 true.

给定数组 A = [1,2,4], 返回 false.

分析:肯定还是动态规划问题,而已可以知道dp[i]肯定和dp[i-1],dp[i-2]有关,注意根据样例的情况,应该是轮流从左拿走1或2个硬币,用dp[i] 表示到第i个硬币时使得先手数量最大的状态。

代码:

class Solution {
public:
    /**
     * @param values: a vector of integers
     * @return: a boolean which equals to true if the first player will win
     */
    bool firstWillWin(vector<int> &values) {
        // write your code here
        int n = values.size();
        if(n<=2)
            return true;
        vector<pair<int,int> > dp(n,make_pair(0,0));
        dp[n-1] = make_pair(values[n-1],0);
        dp[n-2] = make_pair(values[n-1]+values[n-2],0);
        for(int i=n-3;i>=0;i--)
        {
            pair<int,int> p1 = dp[i+1];
            pair<int,int> p2 = dp[i+2];
            if(p1.second+values[i]>p2.second+values[i]+values[i+1])
            {
                dp[i] = make_pair(p1.second+values[i],p1.first);
            }
            else
            {
                dp[i] = make_pair(p2.second+values[i]+values[i+1],p2.first);
            }
        }
        return dp[0].first>dp[0].second;
    }
};


LintCode-硬币排成线 II

标签:面试   lintcode   

原文地址:http://blog.csdn.net/wangyuquanliuli/article/details/45933575

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!