`
haifengyulu
  • 浏览: 14835 次
  • 性别: Icon_minigender_1
  • 来自: 杭州
最近访客 更多访客>>
社区版块
存档分类
最新评论

Leetcode Candy

阅读更多

题目: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;
    }
}

 

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics