【Leetcode】1686. 石子游戏 VI

题目

题目链接
Alice 和 Bob 轮流玩一个游戏，Alice 先手。

一堆石子里总共有 n 个石子，轮到某个玩家时，他可以 移出 一个石子并得到这个石子的价值。Alice 和 Bob 对石子价值有 不一样的的评判标准 。双方都知道对方的评判标准。

给你两个长度为 n 的整数数组 aliceValues 和 bobValues 。aliceValues[i] 和 bobValues[i] 分别表示 Alice 和 Bob 认为第 i 个石子的价值。

所有石子都被取完后，得分较高的人为胜者。如果两个玩家得分相同，那么为平局。两位玩家都会采用 最优策略 进行游戏。

请你推断游戏的结果，用如下的方式表示：

• 如果 Alice 赢，返回 1 。
• 如果 Bob 赢，返回 -1 。
• 如果游戏平局，返回 0 。

示例1：
输入：aliceValues = [1,3], bobValues = [2,1]
输出：1
解释：
如果 Alice 拿石子 1 （下标从 0开始），那么 Alice 可以得到 3 分。
Bob 只能选择石子 0 ，得到 2 分。
Alice 获胜。

示例2：
输入：aliceValues = [1,2], bobValues = [3,1]
输出：0
解释：
Alice 拿石子 0 ， Bob 拿石子 1 ，他们得分都为 1 分。
打平。

示例3：
输入：aliceValues = [2,4,3], bobValues = [1,6,7]
输出：-1
解释：
不管 Alice 怎么操作，Bob 都可以得到比 Alice 更高的得分。
比方说，Alice 拿石子 1 ，Bob 拿石子 2 ， Alice 拿石子 0 ，Alice 会得到 6 分而 Bob 得分为 7 分。
Bob 会获胜。

提示：

• n == aliceValues.length == bobValues.length
• 1 <= n <= 10⁵
• 1 <= aliceValues[i], bobValues[i] <= 100

思路

当一个人拿走石头时，他们不仅会得到分数，而且还会从另一个玩家那里拿走分数。所以我们这里可以使用贪心思想，贪婪地选择具有最大 aliceValues[i] + bobValues[i] 的石头。然后单独计算选择之后每个人的分数，最后再进行比较即可。

代码

class Solution {
public:
    int stoneGameVI(vector<int>& aliceValues, vector<int>& bobValues) {
        vector<pair<int, int>> v;
        int n = aliceValues.size();
        for(int i = 0; i < n; i ++) 
        {
            v.push_back({aliceValues[i], bobValues[i]});
        }
        auto cmp = [&](pair<int, int> p1, pair<int, int> p2) -> bool {
            return p1.first + p1.second > p2.first + p2.second;
        };
        sort(v.begin(), v.end(), cmp);
        int sum1 = 0, sum2 = 0;
        for(int i = 0; i < n; i ++)
        
        {
            if(i % 2 == 0) sum1 += v[i].first;
            else sum2 += v[i].second;
        }
        if(sum1 > sum2) return 1;
        if(sum1 < sum2) return -1;
        return 0;
    }
};

结果