题意:
N个孩子站成一排,给每个人设定一个权重(已知)。按照如下的规则分配糖果: (1)每个孩子至少分得一颗糖果 (2)权重较高的孩子,会比他的邻居获得更多的糖果。
问:总共最少需要多少颗糖果?
1 2 2 竟然是4!!!
因为第三个的权值没有比第二个权值大!
思路:
设数组A1,保存前向遍历的结果,设数组A2,保存后向遍历的结果;
1、与前面的邻居比较,前向遍历权重数组ratings,如果ratings[i]>ratings[i-1],则A1[i]=A1[i-1]+1;
2、与后面的邻居比较,后向遍历权重数组ratings,如果ratings[i]>ratings[i+1],则A2[i]=A2[i+1]+1;
3、每个孩子分到的糖果比前后邻居多,为A1、A2的较大者,即A[i]=max(A1[i],A2[i]);
4、对A求和即为最少需要的糖果。
class Solution {
public:
int candy(vector<int>& ratings) {
int n=ratings.size();
int sum=0;
vector<int> nums(n,1);
vector<int> for_nums(n,1);
vector<int> back_nums(n,1);
for(int i=1;i<n;i++){
if(ratings[i]>ratings[i-1])
for_nums[i]=for_nums[i-1]+1;
}
for(int i=n-2;i>=0;i--){
if(ratings[i]>ratings[i+1])
back_nums[i]=back_nums[i+1]+1;
}
for(int i=0;i<n;i++){
nums[i]=max(for_nums[i],back_nums[i]);
sum+=nums[i];
}
return sum;
}
};
不知道为什么是hard!