题目描述
给你一个下标从 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
算法1
精简下描述:
条件1 每天至少吃1颗糖
条件2 所有type必须从0开始依次吃糖
条件3 每天最多吃dailyCapi可糖
那么根据条件就可以将糖果按照type和数目做成一个数标轴
为了方便的查询某type之前有多少颗糖 使用了前缀和
C++ 代码
class Solution {
public:
long long preSum[100010];
vector<bool> ans;
bool Check( const vector<int>& candiesCount, vector<int> q) {
int type = q[0]; int daylimit = q[1]; int eatPerDay = q[2];
//每天吃1颗的情况下 daylimit+1 必须小于等于 type的糖果的最后一颗糖
if (daylimit + 1 > preSum[type]) return false;
//每天吃最大颗数目的情况下 (daylimit+1)*eatPerDay 必须大于等 type糖果的最开始一颗糖
if (type == 0) return true;
else if (((daylimit + 1)*1LL *eatPerDay) <= preSum[type - 1]) return false;
return true;
}
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
//前缀和
preSum[0] = candiesCount[0];
for (int i = 1; i < candiesCount.size();i++) {
preSum[i] = preSum[i - 1] + candiesCount[i];
}
for (int i = 0; i < queries.size(); i++) {
ans.push_back(Check(candiesCount, queries[i]));
}
return ans;
}
};