文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給你一個下標從 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 <= 10^5 1 <= candiesCount[i] <= 10^5 1 <= queries.length <= 10^5 queries[i].length == 3 0 <= favoriteTypei < candiesCount.length 0 <= favoriteDayi <= 10^9 1 <= dailyCapi <= 10^9

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/can-you-eat-your-favorite-candy-on-your-favorite-day
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 見注釋，前綴和思想

class Solution { public:vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& q) {vector<long long> presum(candiesCount.size());for(int i = 0 ;i < candiesCount.size(); ++i)presum[i] = candiesCount[i];for(int i = 1; i < candiesCount.size(); i++) {presum[i] += presum[i-1];//前綴和}int n = q.size();vector<bool> ans(n, false);for(int i = 0; i < n; i++){int idx = q[i][0];// 要吃的類型int day = q[i][1];// 前面要吃多少天int eat = q[i][2];//每天最多吃多少long long l = (idx > 0 ? (presum[idx-1]+1) : 1), r = presum[idx];// l, r 需要吃到 [l, r] 這個范圍內才行long long L = day*1LL+1, R = (day+1LL)*eat;// L, R 最少，最多能吃的范圍// 兩者有交集 即可if((l >= L && l <= R)||(r >= L && r <= R))ans[i] = true;else if((L >= l && L <= r)||(R >= l && R <= r))ans[i] = true;}return ans;} };

372 ms 118.1 MB C++

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 1744. 你能在你最喜欢的那天吃到你最喜欢的糖果吗？（前缀和）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。