[LeetCode]—Candy 分糖果问题

Candy

 

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:

• Each child must have at least one candy.
• Children with a higher rating get more candies than their neighbors.

题意分析：

                其实只要左右不比他大，就可以只分1个。如果左右比他大，左右就多分一个。这样的结果就是最省的。

                难点在于：例如，[5,4,3,2,4] 虽然知道5比两边的大，但怎么判断到底“纵深是多少（波谷在哪）”应该大几个合适。如果一直苦于搞清楚这个问题，那就进入“死胡同”了。

              其实题目的描述本身就是一个很好的方案：初始化将每个人的糖果数都初始化为1。每次遍历只考虑左右一边，即从左向右遍历，如果i>i-1 则  candy[i]=candy[i-1]；再从右向左遍历一次，如果i>i+1 并且 candy[i]<=candy[i+1] 则 cand[i]=candy[i+1]+1; 这样的两次遍历，左右两边都顾及到了。最后将candy[i]相加求和就好了。

代码：

class Solution{
public:
    int candy(vector<int> &ratings){
      int n=ratings.size();
      if(n==1) return 1;
      int i,sum=0;
      vector<int> candy(n,1);  //先将每人的值都设为1

      //从前向后扫描
     for(i=1;i<n;i++){
        if(ratings[i]>ratings[i-1])candy[i]=candy[i-1]+1;
     }

     //从后向前扫描
     for(i=n-2;i>=0;i--){
        if(ratings[i]>ratings[i+1] && candy[i]<=candy[i+1])candy[i]=candy[i+1]+1;
     }

    for(i=0;i<n;i++)
        sum+=candy[i];
    
    return sum;
    }
};