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

最新推荐文章于 2021-12-09 11:13:17 发布

UQI-LIUWJ

最新推荐文章于 2021-12-09 11:13:17 发布

阅读量90

点赞数

分类专栏： leetcode 文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/qq_40206371/article/details/117441275

版权

leetcode 专栏收录该内容

131 篇文章 0 订阅

订阅专栏

1 题目描述

2 解题思路

我们把所有类型的糖果按照所属类型从小到大排成一队，每天就从前向后依次拿糖果。

那么对于query中的第i个元素 $[favoriteType_i, favoriteDay_i, dailyCap_i]$ ，我们最少已经吃了 $1*(favoriteDay_i +1)$ 颗糖（因为我们是从第0天开始吃的），最多吃了 $dailyCap_i*(favoriteDay_i +1)$ 颗糖。所以只要在这个区间内，有 $favoriteType_i$ 这个类型的糖，那么就满足条件。

而 $favoriteType_i$ 这个类型的糖的下标满足什么条件呢？应该是前一种糖都吃完了，自己吃到最后一颗糖的时候，也就是：

所以只要这两个区间有交集，那么这个query就是True的，否则这个query就是False的

class Solution:
    def canEat(self, candiesCount: List[int], queries: List[List[int]]) -> List[bool]:
        type_sum=[candiesCount[0]]
        for i in candiesCount[1:]:
            type_sum.append(type_sum[-1]+i)
        ret=[]
        for i in queries:
            favor_type=i[0]
            favor_day=i[1]
            daily_cap=i[2]

            #吃糖区间
            eat_min=favor_day+1
            eat_max=(favor_day+1)*daily_cap

            #这个类型的糖的区间(第一类要特判)
            if(favor_type==0):
                sugar_min=0
            else:
                sugar_min=type_sum[favor_type-1]+1
            sugar_max=type_sum[favor_type]

            if(eat_min>sugar_max or eat_max<sugar_min):
                ret.append(False)
            else:
                ret.append(True)
        return(ret)