【LeetCode每日一题】1744. 你能在你最喜欢的那天吃到你最喜欢的糖果吗？

输入：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 的糖果。

输入：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]

class Solution {
public:
    vector<long long> sum;
    bool check(vector<int>& candiesCount, vector<int>& q) {
        int favoriteType = q[0], favoriteDay = q[1], dailyCap = q[2];
        // 第favoriteType类型的前缀和
        int l = sum[favoriteType + 1];
        // 每天吃一颗糖 
        if (favoriteDay + 1 > sum[favoriteType + 1]) return false;
        // 吃最多糖
        if (sum[favoriteType] >= 1ll*(favoriteDay + 1) * dailyCap) return false;
        return true;
    }
    vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
        int n = candiesCount.size();
        sum = vector<long long>(n + 1, 0);
        for (int i = 1; i <= n; i++) {
            sum[i] = sum[i - 1] + candiesCount[i - 1];
        }
        
        vector<bool> ans;
        for (auto q : queries)
            ans.push_back(check(candiesCount, q));
        return ans;
    }
};