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

这是我参与更文挑战的第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. 计算最快速度吃到favoriteType需要多少天(一天吃dailyCap颗)
  2. 计算最慢速度吃到favoriteType需要多少天(一天吃一颗)
  3. 处于这两者之间的天数,都是我们可到达的天数

为了加快速度我们可以,维护一个前缀和,记录下吃到favoriteType糖果之前,需要吃到多少颗的其他糖果。最短天数和最长天数就是由favoriteType之前有多少颗糖果决定的

  • 一天吃一颗需要sum[favoriteType+1](包含favoriteType的糖果数量)
  • 一天吃dailyCap颗需要sum[favoriteType]/dailyCap+1(不包含favoriteType的糖果数量)

代码

func canEat(candiesCount []int, queries [][]int) []bool {

	bools := make([]bool,len(queries))
	sum := make([]int, len(candiesCount)+1)
	for i := 1; i <= len(candiesCount); i++ {
		sum[i]+=candiesCount[i-1]+sum[i-1]
	}
	for j, query := range queries {

		//中间就算合适的天数就是可以达到的天数
		//因为是从0开始统计的,而我们所需要的天数应该是favoriteDay+1
		favoriteType,favoriteDay,dailyCap:=query[0],query[1]+1,query[2]
		
		
		if favoriteDay>=sum[favoriteType] / dailyCap+1&&favoriteDay<=sum[favoriteType+1]{
			bools[j]=true
		}
	}
	return bools
}
复制代码
© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享