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