LeetCode 5667. 你能在你最喜歡的那天吃到你最喜歡的糖果嗎?(字首和)
阿新 • • 發佈:2021-02-17
技術標籤:LeetCode
文章目錄
1. 題目
給你一個下標從 0 開始的正整數陣列 candiesCount ,其中 candiesCount[i]
表示你擁有的第 i 類糖果的數目。
同時給你一個二維陣列 queries ,其中 queries[i] = [favoriteTypei, favoriteDayi, dailyCapi]
。
你按照如下規則進行一場遊戲:
- 你從第 0 天開始吃糖果。
- 你在吃完 所有 第
i - 1
類糖果之前,不能 吃任何一顆第 i 類糖果。 - 在吃完所有糖果之前,你必須每天 至少 吃 一顆 糖果。
請你構建一個布林型陣列 answer ,滿足 answer.length == queries.length 。
不超過 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阿明),一起加油、一起學習進步!