题目链接: 你能在你最喜欢的那天吃到你最喜欢的糖果吗?
使用前缀和数组:
对于: 7 4 5 3 8
前缀: 0 7 11 16 19 27
最慢吃: 如果要吃到糖果0, 那么day不能操作6, 也就是每天只吃一个, 用七天吃完, 即if(pre[i + 1] <= day) return false;
最快吃: 如果要在第0天吃到糖果2, 那么需要把前边的都吃完, 即每天吃的最少为(pre[type] - pre[0]) / (day + 1) + 1
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
vector<bool> ans;
vector<long long> arr(candiesCount.size() + 1, 0);
for (int i = 1; i <= candiesCount.size(); i++) {
arr[i] = arr[i - 1] + candiesCount[i - 1];
}
// for (int i = 0; i < arr.size(); i++) {
// cout << arr[i] << endl;
// }
for (int i = 0; i < queries.size(); i++) {
int type = queries[i][0];
int day = queries[i][1];
if (day >= arr[type + 1]) {
ans.push_back(false);
continue;
}
int cap = queries[i][2];
int need = (arr[type] - arr[0]) / (day + 1) + 1;
if (cap >= need) ans.push_back(true);
else ans.push_back(false);
}
return ans;
}
};