首页 > 代码库 > OJ--分配糖果

OJ--分配糖果

2024-07-17 20:54:46   213人阅读

题目描述

有n（n<=100000）个小朋友站成一排（编号从0到n-1)，每个小朋友有一个rating值，存放在ratings数组中。老师需要给他们分配糖果，每个小朋友至少需要一颗糖果，对于任意相邻的两个小朋友i和i+1，rating值大的必须比rating值小的分配的糖果多（rating相同的没必要分配一样多的糖果）。

请计算最少需要多少颗糖果，才能完成上述分配。

题目在线练习地址：http://www.itint5.com/oj/#31

这道题最开始的思路是将ratings排序，将其下标顺序存储在sort数组中，但是由于先排序，再更新，时间复杂度太高。

于是放弃排序，而是正序遍历一遍，查看相邻数是否反序，再反序遍历一遍，查看相邻数是否反序。这样在O(n)的时间复杂度下就完成了分配。

public class Candy {
	// 返回最少需要的糖果数
		public long minimalCandies(int[] ratings) {
			int n = ratings.length;
			if(n == 0){
				return 0;
			}
			long sum = 0;
			int[] count = new int[n];
			for(int i = 0; i < n; i++){
				count[i] = 1;
				
			}
			
			for(int i = 0; i < n-1; i++){				
				if(ratings[i]>ratings[i+1]){
					if(count[i]<=count[i+1]){
						count[i] = count[i+1]+1;	
					}	
				}else if(ratings[i]<ratings[i+1]){
					if(count[i]>=count[i+1]){
						count[i+1] = count[i]+1;	
					}
				}
			}
			for(int i = n-1; i > 0; i--){
				if(ratings[i]>ratings[i-1]){
					if(count[i]<=count[i-1]){
						count[i]= count[i-1]+1;	
					}
				}else if(ratings[i]<ratings[i-1]){
					if(count[i]>=count[i-1]){
						count[i-1]=count[i]+1;		
					}
				}
			}
			for(int i = 0; i < n; i++){
				sum = sum+ count[i];
			}
			return sum;
		}

}



联系
我们