#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 <= 105possible[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;
}