您的位置:首页 > 编程语言 > C语言/C++

135. Candy

2017-06-08 14:25 106 查看
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.

What is the minimum candies you must give?

N个孩子有不同的等级编号,站成一排等着分糖果。

规定每个孩子至少有一个糖果,等级高的孩子比相邻等级低的孩子分到的糖果要多。

举例说明:输入【2,2】,则每人一个糖果,共需2个;输入【1,2,3】则需要6个糖果;【2,3,2】,则需要4个糖果。

使用贪心算法,先正向遍历,等级高的孩子比左相邻的孩子糖果数加1;同理,再反向遍历,则可以形成金子塔型的糖果分布。

class Solution {
public:
int candy(vector<int>& ratings) {
if(ratings[0]==0)
return 1;
int n = ratings.size();
vector<int> temp(n,1); //temp记录每人手里糖果数量,初始为1

for(int i = 1;i<ratings.size();i++) {     //正向遍历
if (ratings[i]>ratings[i-1]&&temp[i]<=temp[i-1]){
temp[i] = temp[i-1]+1;
}
}
for(int i = n-1;i>=0;i--) {    //反向遍历
if (ratings[i-1]>ratings[i]&&temp[i-1]<=temp[i]){
temp[i-1] = temp[i]+1;
}
}
int sum = 0;
for(int i = 0;i<n;i++) {
sum += temp[i];
}
return sum;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  c++