1744. 你能在你最喜欢的那天吃到你最喜欢的糖果吗？

时间

2021年6月1日（儿童节）

题目

给你一个下标从 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：

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

输入：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 <= 105
1 <= candiesCount[i] <= 105
1 <= queries.length <= 105
queries[i].length == 3
0 <= favoriteTypei < candiesCount.length
0 <= favoriteDayi <= 109
1 <= dailyCapi <= 109

解答

第一次解答

class Solution {
    public boolean[] canEat(int[] candiesCount, int[][] queries) {
        boolean answer[] = null;
        for (int i = 0, day = 0; i < queries.length; i++) {
            int favoriteTypei = queries[i][0];
            int favoriteDayi = queries[i][1];
            int dailyCapi = queries[i][2];
//            控制每天吃糖果的数量
            for (int dailyCap = 1; dailyCap < dailyCapi; dailyCap++) {
//                按照每天吃糖果的数量判断
                for (int j = 0; j < candiesCount.length; day++) {
//                吃糖果
                    candiesCount[j] -= dailyCap;
//                条件满足时
                    if (day == favoriteDayi && j == favoriteTypei) {
                        answer[i] = true;
                        break;
                    }
//                天数已经超过喜欢的天数或者当前吃的糖果类型已经超过喜欢的糖果类型，则直接false
                    if (day > favoriteDayi || j > favoriteTypei) {
                        answer[i] = false;
                        break;
                    }
//                当前类型糖果吃完
                    if (candiesCount[j] <= 0) {
                        ++j;
                    }
                }
            }
        }
    }
}

思路

一般刚看题目都是一头雾水，我都是先暴力算法解一下，然后优化。但是显然这里陷入了一个每天吃糖数是固定的陷阱，题目只是给出了吃糖数的范围是1~dailyCapi ，并没说每天吃dailyCapi ，而上面的代码显然将吃糖数固定了

第二次解答（超时）

class Solution {
 public boolean[] canEat(int[] candiesCount, int[][] queries) {
     boolean answer[] = new boolean[queries.length];
     for (int i = 0; i < queries.length; i++) {
         int favoriteTypei = queries[i][0];
         int favoriteDayi = queries[i][1];
         int dailyCapi = queries[i][2];
         long allShouldMaxEatCandies = 0;
         long allShouldMinEatCandies = 0;
         long canMaxEatCandies = 0;
         long canMinEatCandies = 0;
         for (int j = 0; j <= favoriteTypei; ++j) {
             allShouldMaxEatCandies += candiesCount[j];
             if (j != favoriteTypei) {
                 allShouldMinEatCandies += candiesCount[j];
             }
         }
         allShouldMinEatCandies++;
         canMaxEatCandies = (favoriteDayi + 1) * dailyCapi;
         canMinEatCandies = favoriteDayi + 1;
         //            上面我们已经将达成条件要吃的糖果数的最小与最大值整合成一个范围，如果我们每天能吃的最小值都比达成条件的最大值大则无法迟到或者每天能吃的最大值都比达成条件的最小值小也无法达到条件
         if ((canMinEatCandies <= allShouldMinEatCandies && canMaxEatCandies >= allShouldMaxEatCandies) || (canMinEatCandies >= allShouldMinEatCandies && canMinEatCandies <= allShouldMaxEatCandies) || (canMaxEatCandies >= allShouldMinEatCandies && canMaxEatCandies <= allShouldMaxEatCandies)) {
             answer[i] = true;
         }
     }
     return answer;
 }
}

思路

在第一次解答遇到困难之后，经过思考发现将问题转化成一个比较范围的问题更容易解决。
显然，我们可以根据上面的值得出以下值
1.每天最少能吃多少(canMinEatCandies)：显然每天吃一颗糖就是最少，且要在喜欢的那天吃的最少的糖，那么糖的个数应该就是喜欢的那天的天数+1（天数从0开始算）
2.每天最多能吃多少(canMaxEatCandies)：而每天吃最多，吃到喜欢的天数，显然就是喜欢的天数+1（天数从0开始算）乘以每天吃的最大值（dailyCapi）
注意：通过上面我们已经得出了一个每天能吃的糖数的范围
3.要在喜欢的那天吃到喜欢的糖应该最少吃的数（allShouldMinEatCandies）：这个显然就是将喜欢的类型的糖之前类型的糖的数量累加起来直到喜欢类型的糖之前的类型然后再加+1
4.要在喜欢的那天吃到喜欢的糖应该最多吃的数（allShouldMaxEatCandies）：这个显然就是从第一种糖一直累加到喜欢的类型的糖即可
注意：通过上面我们又得出一个要在喜欢的那天吃喜欢的糖则要吃糖数的范围
这里，我们只需要比较上面的两个范围（每天能吃糖数）vs（每天要吃糖数）

犯的错误

错误1（整型溢出）：原来我将allShouldMaxEatCandies，allShouldMinEatCandies，canMaxEatCandies，canMinEatCandies变量类型置为int，而int的取值范围时2的-32次方到2的31次方（大约21亿），显然当你每天吃840227137颗糖，喜欢第244597天时则最大吃糖数就是205,517,037,028,789（二百多万亿）而int超过范围就会变成负数，导致出错，而long类型的范围是2的-64次方到2的63次方（大约18446744073709551615这么多，我数不过来）
错误2（计算超时）:因为在for循环中多用了一次for循环，直接超时

第三次解答（瞅了一眼官方解法,改了改，和我思想一样嘛！）

class Solution {
    public boolean[] canEat(int[] candiesCount, int[][] queries) {
        boolean answer[] = new boolean[queries.length];
//      遍历出第N天需要吃的糖果数，并存储
        long[] theDayNeedEatCandies = new long[candiesCount.length];
        theDayNeedEatCandies[0] = candiesCount[0];
        for (int j = 1; j < candiesCount.length; ++j) {
            theDayNeedEatCandies[j] = theDayNeedEatCandies[j-1] + candiesCount[j];
        }
//        吃糖
        for (int i = 0; i < queries.length; i++) {
            int favoriteTypei = queries[i][0];
            int favoriteDayi = queries[i][1];
            int dailyCapi = queries[i][2];
            //            设置能吃的范围
            long canMaxEatCandies = (long) (favoriteDayi + 1) * dailyCapi;
            long canMinEatCandies = favoriteDayi + 1;
            //            设置要吃的范围
            long allShouldMinEatCandies = favoriteTypei == 0?1 : theDayNeedEatCandies[favoriteTypei-1]+1;
            long allShouldMaxEatCandies = theDayNeedEatCandies[favoriteTypei];
//            判断能吃的范围是否在要吃的范围之内
            answer[i] = !(canMinEatCandies>allShouldMaxEatCandies || canMaxEatCandies < allShouldMinEatCandies);
        }
        return answer;
    }
}

思路

思路还是第二次的解法，只是优化了for循环减少嵌套，而使用了一个theDayNeedEatCandies数组直接统计第n天要吃的糖果数
另外，这个判断能吃的范围是否在要吃的范围之内也得到了一些优化

注意

六一节快乐~