力扣刷題——3096.得到更多分數的最少關卡數目

SuzumiyaYui發表於2024-08-18

根據題意,假如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;
    }
};

相關文章