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

地址 https://leetcode-cn.com/problems/can-you-eat-your-favorite-candy-on-your-favorite-day/

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

算法1
精简下描述:
条件1 每天至少吃1颗糖
条件2 所有type必须从0开始依次吃糖
条件3 每天最多吃dailyCapi可糖
那么根据条件就可以将糖果按照type和数目做成一个数标轴

为了方便的查询某type之前有多少颗糖 使用了前缀和

class Solution {
public:
    long long  preSum[100010];
    vector<bool> ans;
    bool Check( const vector<int>& candiesCount, vector<int> q) {
        int type = q[0]; int daylimit = q[1]; int eatPerDay = q[2];

        //每天吃1颗的情况下 daylimit+1  必须小于等于 type的糖果的最后一颗糖
        if (daylimit + 1 > preSum[type]) return false;

        //每天吃最大颗数目的情况下 (daylimit+1)*eatPerDay 必须大于等 type糖果的最开始一颗糖
        if (type == 0) return true;
        else if (((daylimit + 1)*1LL *eatPerDay) <= preSum[type - 1]) return false;

        return true;
    }

    vector<bool> canEat(vector<int>& candiesCount, vector<vector<int>>& queries) {
        //前缀和
        preSum[0] = candiesCount[0];
        for (int i = 1; i < candiesCount.size();i++) {
            preSum[i] = preSum[i - 1] + candiesCount[i];
        }

        for (int i = 0; i < queries.size(); i++) {
            ans.push_back(Check(candiesCount, queries[i]));
        }

        return ans;
    }
};

 

posted on 2021-02-04 12:26  itdef  阅读(154)  评论(0编辑  收藏  举报

导航