【leetcode】135.（Hard）Candy

最新推荐文章于 2025-01-04 10:10:24 发布

原创最新推荐文章于 2025-01-04 10:10:24 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一种高效解决糖果分配问题的算法，通过三次遍历确保每个孩子得到的糖果数量与其等级相匹配，避免了多次迭代的复杂度。首次遍历从左到右，确保等级高的孩子比左侧的孩子糖果多；第二次遍历从右到左，确保等级高的孩子比右侧的孩子糖果多；最后统计总糖果数。整体时间复杂度为O(n)，空间复杂度也为O(n)。

解题思路：
3次遍历：
第一次从左向右：自己等级比左边高，则自己的糖是[左边糖数+1]，否则自己一颗糖
第二次从右向左：如果自己等级比右边高，但是糖没有右边多，则自己糖数是[右边糖数+1]
第三次：统计所有糖果数
时间复杂度O(n) 空间复杂度：O(n)

提交代码：

class Solution {
    public int candy(int[] ratings) {
    	if(ratings.length==0)	return 0;
    	int[] candys=new int[ratings.length];
    	candys[0]=1;
    	
    	for(int i=1;i<ratings.length;i++) {
    		if(ratings[i]>ratings[i-1])	candys[i]=candys[i-1]+1;
    		else candys[i]=1;
    	}
    	
    for(int i=ratings.length-2;i>=0;i--) {
    	if(ratings[i]>ratings[i+1]&&candys[i]<=candys[i+1])	
    		candys[i]=candys[i+1]+1;
    }
    
    int res=0;
    for(int i=0;i<candys.length;i++)
    	res+=candys[i];
    
    return res;
    }
}