给你一个下标从 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]
/**
* @param {number[]} candiesCount
* @param {number[][]} queries
* @return {boolean[]}
*/
// 1、前缀和
// 2、能在 最喜欢的那天 吃到 最喜欢的糖果
// 3、需要满足两个条件
// 4、上边界:我们现在以最大数量去吃糖,那么,在最喜欢那天,一共是吃掉 Dayi * Capi
// 块糖果,该值应该大于 favoriteTypei(但是不包含favoriteTypei) 之前的糖果数量之和。
// 5、下边界:我们现在每天 吃一块糖,吃到最喜欢那天,一共是吃掉 Dayi + 1(因为从第 0 天开始吃糖果) ,该值应该小于 favoriteTypei(但是不包含favoriteTypei) 之前的糖果数量之和。
var canEat = function(candiesCount, queries) {
const n = candiesCount.length,len = queries.length;
// 计算前缀和
let pre = new Array(n + 1);
pre[0] = 0;
for(let i = 0;i < n;i++){
pre[i + 1] = pre[i] + candiesCount[i];
}
let res = [];
for(let [ft,fd,dc] of queries){
// 计算上下边界
let low = fd + 1;
let high = dc * (fd + 1);
// true条件判断
if(pre[ft + 1] >= low && pre[ft] < high){
res.push(true);
}else{
res.push(false);
}
}
return res;
};