#medium #前缀和
给你一个长度为 n
的二进制数组 possible
。
Alice 和 Bob 正在玩一个有 n
个关卡的游戏,游戏中有一些关卡是 困难 模式,其他的关卡是 简单 模式。如果 possible[i] == 0
,那么第 i
个关卡是 困难 模式。一个玩家通过一个简单模式的关卡可以获得 1
分,通过困难模式的关卡将失去 1
分。
游戏的一开始,Alice 将从第 0
级开始 按顺序 完成一些关卡,然后 Bob 会完成剩下的所有关卡。
假设两名玩家都采取最优策略,目的是 最大化 自己的得分,Alice 想知道自己 最少 需要完成多少个关卡,才能获得比 Bob 更多的分数。
请你返回 Alice 获得比 Bob 更多的分数所需要完成的 最少 关卡数目,如果 无法 达成,那么返回 -1
。
注意,每个玩家都至少需要完成 1
个关卡。
提示:
2 <= n == possible.length <= 105
possible[i]
要么是0
要么是1
。
思路
我们可以通过前缀和可以计算出每通过一关后积累的分数,以及通过剩余关数的分数,再通过比较 a[possibeleSize - 1] - a[i] < a[i]
即 2 * a[i] > a[possibleSize - 1]
即可
由于两人都要有参与关卡,所以 i < possibleSize - 1
int minimumLevels(int* possible, int possibleSize) {
int a[100010];
//生成前缀和数组
a[0] = possible[0] == 0 ? -1 : 1;
for (int i = 1; i < possibleSize; i++) {
if (possible[i] == 0) {
a[i] = a[i - 1] - 1;
} else {
a[i] = a[i - 1] + 1;
}
}
for (int i = 0; i < possibleSize - 1; i++) {
if (2 * a[i] > a[possibleSize - 1])
return i + 1;
}
return -1;
}