分糖果（条件分配）

题目：老师想给孩子们分发糖果，有N个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。你需要按照以下要求，帮助老师给这些孩子分发糖果：

每个孩子至少分配到 1 个糖果。

相邻的孩子中，评分高的孩子必须获得更多的糖果。

那么这样下来，老师至少需要准备多少颗糖果呢？

示例

输入: [1,0,2]

输出: 5

解释: 可以分别给这三个孩子分发 2、1、2 颗糖果。

#include<iostream>
using namespace std;

int main(){
	int n;
	cin>>n;
	int a[n];
	int b[n];//每个人的糖果数 
	int min_i=0;
	for(int i=0;i<n;i++){
		cin>>a[i];
		if(a[i]<a[min_i]) min_i=i;
	}
	b[min_i]=1;//评分最小的孩子给1个 
	for(int i=min_i+1;i<n;i++){
		if(a[i]>a[i-1]) b[i]=b[i-1]+1;
		else if(a[i]==a[i-1]) b[i]=b[i-1];
		else b[i]=1;
	}
	for(int i=min_i-1;i>=0;i--){
		if(a[i]>a[i+1]) b[i]=b[i+1]+1;
		else if(a[i]==a[i+1]) b[i]=b[i+1];
		else b[i]=1;
	} 
	//求和
	int sum=0;
	for(int i=0;i<n;i++){
		sum+=b[i]; 
	} 
	cout<<sum<<endl;
	return 0;
}