LeetCode 1744 你能在你最喜欢的那天吃到你最喜欢的糖果吗？[前缀和] HERODING的LeetCode之路

Posted 2021-06-15 HERODING23

解题思路：
问题不难，但是理解题意真的很费劲！解决该题的关键还是在于计算前缀和，以及边界的控制。首先计算出前i类糖果的总数，之后遍历queries数组，定义并计算当前状态下最多以及最少能吃的糖果数，与吃到第i类糖果的前缀和前后边界进行对比，如果最多糖果数量比i类糖果前缀和前边界数目还小，或者最少糖果数比i类糖果前缀和后边界还大，那说明不可能吃到i类糖果，反之可以，代码如下：

class Solution {
public:
    vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
        vector<long long> count(candiesCount.size());
        count[0] = candiesCount[0];
        // 计算前缀和，即前i类糖果的总数
        for(int i = 1; i < candiesCount.size(); i ++) {
            count[i] = count[i - 1] + candiesCount[i];
        }
        vector<bool> judge;
        for(auto& q : queries) {
            // 下界
            long long low1 = q[1] + 1;
            long long low2 = (q[0] == 0 ? 1 : count[q[0] - 1] + 1);
            // 上界
            long long high1 = (long long)q[2] * (q[1] + 1);
            long long high2 = count[q[0]];
            if(!(low1 > high2 || high1 < low2)) {
                judge.push_back(true);
            } else {
                judge.push_back(false);
            }
        }
        return judge;
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/can-you-eat-your-favorite-candy-on-your-favorite-day/solution/cqian-zhui-he-by-heroding-wd61/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/