给你一个下标从 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]
代码
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
int n = candiesCount.size();
vector<long long> total(n+1);
total[0] = 0;
//total是第i类糖之前总糖果数
for(int i = 1; i <= n; i++){
total[i] = total[i-1] + candiesCount[i-1];
}
long long Most = 0;
vector<bool> answer;
//前面糖果太多
for(const auto& query : queries){
long long favoriteDay = query[1];
long long dailyCap = query[2];
//最多能吃几个糖
Most = (favoriteDay+1) * dailyCap;
if(Most <= total[query[0]]){
answer.push_back(false);
}
//天数太多,每天至少吃一个,在到那天之前就把想吃的吃完了
else if(query[1]+1 > total[query[0] + 1])
{
answer.push_back(false);
}
else
{
answer.push_back(true);
}
}
return answer;
}
};
时间复杂度:O(n)
空间复杂度:O(n)
这道题目题目比较长,但是实际上就是每天至少吃一个糖,最多吃query[2]个糖,然后要保证在query[1]天的时候(从第0天开始计),要把第 i 类糖之前的糖吃完。先定义一个total数组用来储存第i类糖之前的糖果总数,然后列出两种无法在特定日期吃到特定糖的情况。一种是每天吃query[2]个糖都无法把第i类糖之前的糖吃完,第二种是即使每天吃一个,但是在指定日期来临前,想吃的糖已经被吃完两种情况。如果出现这两种情况,则为false,否则为true。