【LeetCode】第1174题——你能在你最喜欢的那天吃到你最喜欢的糖果吗（难度：中等）

题目描述

给你一个下标从 0 开始的正整数数组 candiesCount ，其中 candiesCount[i] 表示你拥有的第 i 类糖果的数目。同时给你一个二维数组 queries ，其中 queries[i] = [favoriteTypei, favoriteDayi, dailyCapi] 。

你按照如下规则进行一场游戏：

你从第 0 天开始吃糖果。
你在吃完 所有 第 i - 1 类糖果之前，不能 吃任何一颗第 i 类糖果。
在吃完所有糖果之前，你必须每天 至少 吃 一颗 糖果。

请你构建一个布尔型数组 answer ，满足 answer.length == queries.length 。answer[i] 为 true 的条件是：在每天吃 不超过 dailyCapi 颗糖果的前提下，你可以在第 favoriteDayi 天吃到第 favoriteTypei 类糖果；否则 answer[i] 为 false 。注意，只要满足上面 3 条规则中的第二条规则，你就可以在同一天吃不同类型的糖果。

请你返回得到的数组 answer 。

1. 示例 1：
   输入：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 的糖果。

2. 示例 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]

提示：
1 <= candiesCount.length <= 10⁵
1 <= candiesCount[i] <= 10⁵
1 <= queries.length <= 10⁵
queries[i].length == 3
0 <= favoriteTypei < candiesCount.length
0 <= favoriteDayi <= 10⁹
1 <= dailyCapi <= 10⁹

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/can-you-eat-your-favorite-candy-on-your-favorite-day
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

利用前缀和数组sum[]，长度为candiesCount的长度+1，sum[i]代表吃过sum[i]个糖果后才能吃第i种糖果

如果能在最喜欢的那天吃到最喜欢的糖果（注意是从第0天开始吃的），那么最晚吃到第i类糖果所经历的天数为queries[i][1]+1，即以每天吃一个糖果的速率。最早能吃到第i类糖果所经历的天数为sum[queries[i][0]] / queries[i][2] + 1（因为“/”为向下取整，因此+1）。

这样，只要最短天数不大于最长天数（如果大于，就代表以最快的速率，也不能在最喜欢那天吃到最喜欢的糖果），且最长天数要不大于sum[queries[i][0]+1]（如果大于，就代表以一天一颗糖果的速率，在最喜欢那天之前就已经把最喜欢的糖果吃完了）

代码详解

class Solution {
    public boolean[] canEat(int[] candiesCount, int[][] queries) {
        boolean[] ans = new boolean[queries.length];
        long[] sum = new long[candiesCount.length+1]; // 前缀和
        // 构建前缀和
        for(int i = 1; i < sum.length; ++i) {
            sum[i] = sum[i-1] + candiesCount[i-1];
        }
        for(int i = 0; i < queries.length; ++i) {
            int max = queries[i][1] + 1; // 最晚的天数
            long min = sum[queries[i][0]] / queries[i][2] + 1; // 最早的天数
            ans[i] = (min <= max && max <= sum[queries[i][0]+1]); // 判断
        }
        return ans;
    }
}

注意点

• 本题注意是用除法算的天数，如果反过来使用乘法算糖果数会导致int越界，long也能越界，会带来不必要的麻烦