Medium
Can You Eat Your Favorite Candy on Your Favorite Day? — C++
Full explanation · Time O(n) · Space O(n)
// Time: O(n)
// Space: O(n)
class Solution {
public:
vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
vector<int64_t> prefix(size(candiesCount) + 1);
for (int i = 0; i < size(candiesCount); ++i) {
prefix[i + 1] = prefix[i] + candiesCount[i];
}
vector<bool> result;
for (const auto& q : queries) {
result.emplace_back(prefix[q[0]] / q[2] < q[1] + 1 && q[1] + 1 <= prefix[q[0] + 1] / 1);
}
return result;
}
};