OJ--分配糖果

題目描述

有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;}}