leetcode 1744.你能在你最喜欢的那天吃到你最喜欢的糖果吗?
题干
给你一个下标从 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
知识点&算法
上当了,吃糖是从第0天开始吃的。
用前缀和保存前i类共有多少颗糖,然后对于每个query,先算出吃到目标天数(含目标天)最少能吃几颗糖,最多能吃几颗糖
如果最少能吃的糖小于等于目标类及此前糖总数,最多能吃的糖大于目标类之前糖的总数,那就说明能在该天吃到。
题解
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
int n = candiesCount.size(), m = queries.size();
long long type = -1, limit = -1, day = -1, minCandies = -1, maxCandies = -1, preSum[100005] = {0};
vector<bool> res(m);
for(int i = 1 ; i <= n ; ++i){
preSum[i] = preSum[i-1] + candiesCount[i-1];
}
for(int i = 0 ; i < m ; ++i){
type = queries[i][0];
day = queries[i][1];
limit = queries[i][2];
minCandies = day + 1;
maxCandies = limit * (day + 1);
if(minCandies <= preSum[type+1] && maxCandies > preSum[type]) res[i] = true;
else res[i] = false;
}
return res;
}
};