「这是我参与11月更文挑战的第26天,活动详情查看:2021最后一次更文挑战」
题目要求
题目描述
给你一个下标从 0 开始的正整数数组 candiesCount ,其中 candiesCount[i] 表示你拥有的第 i 类糖果的数目。同时给你一个二维数组 queries ,其中 queries[i] = [favoriteTypei, favoriteDayi, dailyCapi] 。
你按照如下规则进行一场游戏:
你从第 0 天开始吃糖果。 你在吃完 所有 第 i - 1 类糖果之前,不能 吃任何一颗第 i 类糖果。 在吃完所有糖果之前,你必须每天 至少 吃 一颗 糖果。 请你构建一个布尔型数组 answer ,用以给出 queries 中每一项的对应答案。此数组满足:
answer.length == queries.length 。answer[i] 是 queries[i] 的答案。 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
解题思路
对示例进行一遍查看之后,我们很容易就能得出一个思路。假设我们需要在第n
天吃到第k
类糖果,且每天最多能吃的糖果个数为m
。那么只需要满足下面两个条件就能使结果为true
:
- 假设每天只吃一个糖果,那么第
n-1
天吃掉的糖果的总个数(即为n-1
)小于前k
类糖果的总个数。 - 假设每天吃
m
个糖果,那么第n-1
天吃掉的糖果的总个数(即为(n-1)*m
)大于前k-1
类糖果的总个数。
根据上述思路,可以得到第一个版本的代码,但是由于dailyCapi
的最大值为10的9次方,因此计算可能会超出范围,因此我们可以换一个方向,只需满足下面两个条件就能使结果为true
:
- 假设每天只吃一个糖果,最多在
前k类糖果总数
的天数前就能吃掉喜欢的糖果。 - 假设每天吃
m
个糖果,那么最少要在(前k-1类糖果总数)/m
的天数后才能吃掉喜欢的糖果。
根据上述思路即可编写出正确题解。
代码
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
vector<int> candy_sum(candiesCount.size()+1);
vector<bool> res;
candy_sum[0] = 0;
for(int i=0; i<candiesCount.size(); i++){
candy_sum[i+1] = candy_sum[i] + candiesCount[i];
}
for(int i=0; i<queries.size(); i++){
/*if(queries[i][1]<=candy_sum[queries[i][0]+1] && queries[i][1]*queries[i][2]>candy_sum[queries[i][0]]){
res.push_back(true);
}else{
res.push_back(false);
}*/
if(candy_sum[queries[i][0]]/queries[i][2] < queries[i][1] && queries[i][1] <= candy_sum[queries[i][0]+1]){
res.push_back(true);
}else{
res.push_back(false);
}
}
return res;
}
};
复制代码