题目:
给你一个下标从 0 开始的正整数数组 candiesCount ,其中 candiesCount[i] 表示你拥有的第 i 类糖果的数目。同时给你一个二维数组 queries 。
其中 queries[i] = [favoriteTypei, favoriteDayi, dailyCapi] 。你按照如下规则进行一场游戏:
1.你从第 0 天开始吃糖果。
2.你在吃完 所有 第 i - 1 类糖果之前,不能 吃任何一颗第 i 类糖果。
3.在吃完所有糖果之前,你必须每天 至少 吃 一颗 糖果。
请你构建一个布尔型数组 answer ,满足 answer.length == queries.length 。answer[i] 为 true 的条件是:在每天吃 不超过 dailyCapi 颗糖果的前提下,你可以在第 favoriteDayi 天吃到第 favoriteTypei 类糖果;否则 answer[i] 为 false 。注意,只要满足上面 3 条规则中的第二条规则,你就可以在同一天吃不同类型的糖果。请你返回得到的数组 answer 。
示例:
输入: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 的糖果。
提示:
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颗糖果,至多吃dailyCapi 颗糖果,这么多天是按顺序从candiesCount这个糖果种类和数量存储容器中取出糖果来吃的,在前面种类糖果吃完前不能吃后面种类,一天可以吃多个种类,只要你前面糖果全部吃完。就是给了你一个queries数组,里面存了好几个条件,在每个条件下判断能否在最喜欢的天数上吃到最喜欢的糖果种类。
那好了,直接对这个queries数组进行遍历,然后依次判断这个返回true还是false不就行了。能否吃到最喜欢的糖果种类? 那好了,首先对candiesCount计算前缀和存到sum数组。怎么才能在最喜欢的天数吃到呢?那么既然知道了天数,也知道了每天能吃的最大最小糖果数量,我们可以得到这个情况下我能得到的mindata和maxdata代表目标天数上能迟到的最大和最小糖果数量,[mindata,maxdata]是一个区间,我们需要保证这个区间和我们目标糖果所在的区间重合即可。如果maxdata比区间左侧还小肯定吃不到。 考虑maxdata小于等于区间右临界点,然后大于等于区间左临界点即可有解。或者maxdata大于区间右临界点,但是mindata只要小于等于区间右临界点就行了! 这样题目就解决了。
然后我们要看下时间复杂度会不会溢出呢?循环queries里面的元素设长度是m。计算前缀和假设长度是n,总的时间复杂度是o(m+n)不会溢出没问题。我们用了额外的sum数组计算前缀和,空间复杂度是o(n)。
然后注意到,现在看题目要看下提示中数据的区间范围,看是否会溢出! 在我们的计算中,dailyCapi *favoriteDayi会溢出int,因为我们知道10^9约等于2的32次幂,但是稍微少一点,他们乘起来这个数字就大了会溢出int,然后sum数组是计算前缀和可能会溢出因为是10的5次方乘以10的五次方可能要10的10次方了。所以这里也要注意。哦,好像在32位机子上long就是int但是64位机子,long=long long ,我是64机子,直接用long即可了。下面附上我的代码:
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
int size=queries.size();
vector<bool> ans(size);
//吃的最多糖果要大于等于临界左侧,小于等于临界右侧;或者吃的最多的糖果大于临界右侧,但是最少糖果小于等于临界右侧
vector<long> sum(candiesCount.size()+1);
for(int i=1; i<=candiesCount.size();i++){
sum[i]=sum[i-1]+candiesCount[i-1]; //计算累加数组
}
for(int i=0; i<queries.size();i++){
long long maxdata=(long long)queries[i][2]*(queries[i][1]+1); //计算的过程中会溢出,给他转换成long
int mindata=1*(queries[i][1]+1);
int target=(queries[i][0]+1); //第target类型的糖果,从1开始计数
//sum[target]右边界 sum[target-1]+1 左边界
if(maxdata<=sum[target] && maxdata>=sum[target-1]+1 || maxdata>sum[target] && mindata<=sum[target] )
ans[i]=true;
else
ans[i]=false;
}
return ans;
}
};
思路2:
刚看了Leetcode的官方题解,和我的思路是一致的,但是换句话说,找这个边界的时候我找的是true的情况,我们也可以找false的情况,除了false其他情况就肯定是true了嘛。false比较好找,就是最小糖果大于有边界或者最大糖果小于左边界,有一个条件满足就不行。
class Solution {
private:
using LL = long long;
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
int n = candiesCount.size();
// 前缀和
vector<LL> sum(n);
sum[0] = candiesCount[0];
for (int i = 1; i < n; ++i) {
sum[i] = sum[i - 1] + candiesCount[i];
}
vector<bool> ans;
for (const auto& q: queries) {
int favoriteType = q[0], favoriteDay = q[1], dailyCap = q[2];
LL x1 = favoriteDay + 1;
LL y1 = (LL)(favoriteDay + 1) * dailyCap;
LL x2 = (favoriteType == 0 ? 1 : sum[favoriteType - 1] + 1);
LL y2 = sum[favoriteType];
ans.push_back(!(x1 > y2 || y1 < x2));
}
return ans;
}
};