【LeetCode】你能在你最喜欢的那天吃到你最喜欢的糖果吗？Java 题解

关注

发布于: 2021 年 06 月 02 日

题目描述

给你一个下标从 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 的糖果。

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

复制代码

思路分析

这个题目题干比较长，需要认真阅读才能理解题意。
分析题意之后，题目核心是分析第 favoriteDay 天能不能吃到 favoriteType 类型糖果。由此，可以得到一个区间范围，最小吃到的糖果数量是 favoriteDay + 1, 最大吃到的糖果数量是（favoriteDay + 1) * dailyCap。在此区间范围的累计糖果数即可满足条件，返回 true;

AC 代码

public class DayCode {    public static void main(String[] args) {        int[] candiesCount = new int[]{7, 4, 5, 3, 8};        int[][] queries = new int[][]{{0, 2, 2}, {4, 2, 4}, {2, 13, 1000000000}};        boolean[] ans = new DayCode().canEat(candiesCount, queries);        System.out.println(Arrays.toString(ans));
        int[] candiesCount1 = new int[]{5, 2, 6, 4, 1};        int[][] queries1 = new int[][]{{3, 1, 2}, {4, 10, 3}, {3, 10, 100}, {4, 100, 30}, {1, 3, 1}};        boolean[] ans1 = new DayCode().canEat(candiesCount1, queries1);        System.out.println(Arrays.toString(ans1));    }
    public boolean[] canEat(int[] candiesCount, int[][] queries) {        int m = candiesCount.length;        long[] sum = new long[m];        sum[0] = candiesCount[0];        for (int i = 1; i < m; ++i) {            sum[i] = sum[i - 1] + candiesCount[i];        }
        int n = queries.length;        boolean[] ans = new boolean[n];        for (int i = 0; i < n; i++) {            int favoriteType = queries[i][0];            int favoriteDay = queries[i][1];            int dailyCap = queries[i][2];
            long x1 = favoriteDay + 1;            long y1 = (long) (favoriteDay + 1) * dailyCap;            long x2 = favoriteType == 0 ? 1 : sum[favoriteType - 1] + 1;            long y2 = sum[favoriteType];
            ans[i] = !(x1 > y2 || y1 < x2);        }
        return ans;    }}

复制代码

总结

上述算法代码的时间复杂度是 O(n), 空间复杂度是 O(1)
在计算糖果数时，采用了前缀和的方法。前缀和是一种重要的预处理，能大大降低查询的时间复杂度。
求前缀和的代码十分经典，熟练掌握可以提升解题效率。
坚持每日一题，加油！

发布于: 2021 年 06 月 02 日阅读数: 6

原文链接:【http://xie.infoq.cn/article/e7437de1432ffaa24ab2e1962】。文章转载请联系作者。

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

发布

暂无评论

创作场景

【LeetCode】你能在你最喜欢的那天吃到你最喜欢的糖果吗？Java 题解

题目描述

思路分析

AC 代码

总结

HQ数字卡

评论