1. 程式人生 > 其它 >Leetcode之你能在你最喜歡的那天吃到你最喜歡的糖果嗎

Leetcode之你能在你最喜歡的那天吃到你最喜歡的糖果嗎

技術標籤:leetcode和機試題leetcode

題目:

給你一個下標從 0 開始的正整數陣列candiesCount,其中candiesCount[i]表示你擁有的第i類糖果的數目。同時給你一個二維陣列queries,其中queries[i] = [favoriteTypei, favoriteDayi, dailyCapi]。

你按照如下規則進行一場遊戲:

你從第0天開始吃糖果。
你在吃完 所有第 i - 1類糖果之前,不能吃任何一顆第 i類糖果。
在吃完所有糖果之前,你必須每天 至少吃 一顆糖果。
請你構建一個布林型陣列answer,滿足answer.length == queries.length 。answer[i]為true的條件是:在每天吃 不超過 dailyCapi顆糖果的前提下,你可以在第favoriteDayi天吃到第favoriteTypei類糖果;否則 answer[i]為 false。注意,只要滿足上面 3 條規則中的第二條規則,你就可以在同一天吃不同型別的糖果。

請你返回得到的陣列answer。

示例 1:

輸入:candiesCount = [7,4,5,3,8], queries = [[0,2,2],[4,2,4],[2,13,1000000000]]
輸出:[true,false,true]
提示:
1- 在第 0 天吃 2 顆糖果(型別 0),第 1 天吃 2 顆糖果(型別 0),第 2 天你可以吃到型別 0 的糖果。
2- 每天你最多吃 4 顆糖果。即使第 0 天吃 4 顆糖果(型別 0),第 1 天吃 4 顆糖果(型別 0 和型別 1),你也沒辦法在第 2 天吃到型別 4 的糖果。換言之,你沒法在每天吃 4 顆糖果的限制下在第 2 天吃到第 4 類糖果。
3- 如果你每天吃 1 顆糖果,你可以在第 13 天吃到型別 2 的糖果。

示例 2:

輸入:candiesCount = [5,2,6,4,1], queries = [[3,1,2],[4,10,3],[3,10,100],[4,100,30],[1,3,1]]
輸出:[false,true,true,false,false]

提示:

1 <= candiesCount.length <= 105
1 <= candiesCount[i] <= 105
1 <= queries.length <= 105
queries[i].length == 3
0 <= favoriteTypei < candiesCount.length
0 <= favoriteDayi <= 109

1 <= dailyCapi <= 109

程式碼:

class Solution {
public:
    vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
         vector<long long> sum={0};
    for(int i=0;i<candiesCount.size();i++){
        sum.push_back(candiesCount[i]+sum.back());
    }
    vector<bool> res;
    for(int i=0;i<queries.size();i++){
        double a =queries[i][1]+1;
        double b=queries[i][2];
        double maxeat=a*b;
        if(queries[i][1]<sum[queries[i][0]+1]&&(maxeat>sum[queries[i][0]])){
            res.push_back(true);
        }else res.push_back(false);

    }
    return res;
    }
};

想法:對於每類糖果,算出sum陣列,遍歷每個query,計算出能吃到query在favorite day中能吃到的糖果的區間,第i天(從0開始)最少是最小是i+1,因為每天至少要吃一個,這個要小於sum[j],防止第j個糖果不夠吃到第i天;第i天計算最多是(i+1)*q[2],按每天吃最多的算,這個最多要超過前一類的sum,即超過sum[j-1],防止即使每天吃最多也吃不到第i類;對於每個query,判斷這兩個條件是否滿足,如果都滿足,那麼返回true;對於很多個queries,那麼自然形成了由true和false組成的query列表。