根據題意,假如alice選擇完成第i關到第j關,那麼bob需要完成第j+1關到第n關,其中i<=j<n。如此可以想到對關卡陣列進行預處理,構建一個字首和陣列,儲存假如從第0關每關都透過的話,到第i關所得到的分數。
透過遍歷一次字首和陣列,能夠得到每個時刻alice得到的分數和bob得到的分數,當alice獲得的分數超過了bob那麼就可以直接輸出結果,假如遍歷完成後仍然沒有alice得分高於bob得分的情況,那麼就輸出-1。
class Solution
{
public:
int minimumLevels(vector<int> &possible)
{
vector<int> preSum;
int totalSum = possible[0] == 1 ? 1 : -1;
preSum.emplace_back(0);
int alicePlay = 1;
for (int i = 1; i < possible.size(); i++)
{
preSum.emplace_back(totalSum);
totalSum += possible[i] == 1 ? 1 : -1;
}
while (alicePlay < preSum.size())
{
if (preSum[alicePlay] > totalSum - preSum[alicePlay])
return alicePlay;
alicePlay++;
}
return -1;
}
};