码迷,mamicode.com
首页 > 其他好文 > 详细

OJ--分配糖果

时间:2014-08-15 23:52:29      阅读:422      评论:0      收藏:0      [点我收藏+]

标签:style   http   color   java   for   ar   div   时间   

题目描述

n(n<=100000)个小朋友站成一排(编号从0n-1),每个小朋友有一个rating值,存放在ratings数组中。老师需要给他们分配糖果,每个小朋友至少需要一颗糖果,对于任意相邻的两个小朋友ii+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;
		}

}


OJ--分配糖果,布布扣,bubuko.com

OJ--分配糖果

标签:style   http   color   java   for   ar   div   时间   

原文地址:http://blog.csdn.net/lands92/article/details/38592951

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!