LeetCode
石子游戏 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.length1 <= n <= 1051 <= aliceValues[i], bobValues[i] <= 100
思路
贪心
思路 get,以后贪心想不出来就想基本情况(只有两个的情况),然后再推式子
代码
C++
class Solution {
public:
int stoneGameVI(vector<int>& aliceValues, vector<int>& bobValues) {
vector<pair<int, int>> mp; // 记录价值和以及下标
int n = aliceValues.size();
for (int i = 0; i < n; i++) {
int dis = aliceValues[i] + bobValues[i];
mp.emplace_back(dis, i);
}
sort(mp.rbegin(), mp.rend()); //从大到小排序
int sum = 0, sum2 = 0;
for (int i = 0; i < n;
i++) { // Alice 和 Bob 轮流玩一个游戏,Alice 先手。
if (i % 2 == 0)
sum += aliceValues[mp[i].second]; // 偶数下标a来取
else
sum2 += bobValues[mp[i].second]; // 奇数下标b来取
}
if (sum == sum2) {
return 0;
} else if (sum > sum2) {
return 1;
} else{
return -1;
}
}
};
Java
class Solution {
public int stoneGameVI(int[] a, int[] b) {
int n = a.length;
Integer[] ids = new Integer[n];
for(int i = 0;i < n ; i++){
ids[i] = i;
}
Arrays.sort(ids,(i,j) -> a[j] + b[j] - a[i] - b[i]);
int diff = 0;
for(int i = 0;i < n; i++){
diff += i % 2 == 0 ? a[ids[i]] : -b[ids[i]];
}
return Integer.compare(diff,0);
}
}