题目:http://oj.leetcode.com/problems/candy/
这道题很多人做了很长时间,从通过率也可以看出来。我的第一想法是从前往后遍历,如果i+1的rating大于i,则i+1的candy等于i的candy+1;如果小于,则从i开始往前遍历,直到处理完所有rating大且candy小于等于的情况,然后继续这个过程。这个思路代码实现起来稍微有点麻烦,而且时间复杂度是O(n平方)。
后者仔细想了一下,其实不用这么复杂,只需从前往后扫描(处理i+1的rating大于i的情况)+从后往前扫描(i的rating大于i+1的情况) 两次遍历即可,具体代码如下:
public class Solution {
public int candy(int[] ratings) {
int[] candies=new int[ratings.length];
Arrays.fill(candies, 1);
for(int i=1;i<ratings.length;i++){
if(ratings[i]>ratings[i-1])
candies[i]=candies[i-1]+1;
}
for(int i=ratings.length-2;i>=0;i--){
if(ratings[i]>ratings[i+1]&&candies[i]<=candies[i+1])
candies[i]=candies[i+1]+1;
}
int sum=0;
for(int i=0;i<ratings.length;i++){
sum+=candies[i];
}
return sum;
}
}