派分糖果
有 N 个孩子站成一排,每个孩子有一个分值。给这些孩子派发糖果,需要满足如下需求:
1、每个孩子至少分到一个糖果
2、分值更高的孩子比他相邻位的孩子获得更多的糖果
求至少需要分发多少糖果?
- 输入描述:
0,1,0
- 输出描述:
4
- 输入示例:
5,4,1,1
- 输出示例:
7
#include<bits/stdc++.h>
using namespace std;
int main(){
int n = 5;
int sugers[n];
sugers[0] = 1;
for(int i = 1; i < n; i++){
if(nums[i] == nums[i-1])
sugers[i] = sugers[i-1];
else if(nums[i] > nums[i-1])
sugers[i] = sugers[i-1] + 1;
else{
sugers[i] = 1;
int now = i;
int pre = now-1;
while(pre >= 0&&nums[pre] >= nums[now]&&sugers[pre]<=sugers[now]){
++sugers[pre];
--now;
--pre;
}
}
int res = 0;
for(int i=0;i<n;i++){
res+=sugers[i];
}
printf("%d\n",res);
}