为N个小孩分糖果所需要的最少糖果数目

有 N 个小孩站成一列，每个小孩手中有一个随机数，按照以下规则分配糖果：

1、如果某个小孩手中的随机数比旁边的大，他分到的糖果数目就要比旁边的多；

2、每个小孩至少得到一颗糖果；

问：需最少准备多少糖果？

#include<iostream>
#include<vector>;
using namespace std;

int candy(vector<int>& ratings) 
{
	int sum = 0;//统计糖果的总数目
	vector<int> v1;//存储每个小孩分配的糖果数目
	if (ratings.size() != 0)
	{
		int num = 1;
		v1.push_back(num);
		if (ratings.size() == 1)
			return 1;
		for (size_t i = 1; i != ratings.size(); ++i)
		{
			if (ratings[i] > ratings[i - 1])
			{
				num++;//如果比前一个小孩的数字大，他得到的糖果数目比前一个多1
			}
			else
			{
				num = 1;//如果没比前一个小孩的数字大，他得到的糖果数目维持为1
			}
			v1.push_back(num);
		}
		sum += num;//统计倒数第一个小孩的糖果数目
		for (size_t i = ratings.size() - 2; i != -1; --i)//统计其余小孩的糖果数目
		{
			if (ratings[i] > ratings[i+1])
			{
				num=v1[i+1]+1;//如果比后一个小孩的数字大，i得到的糖果数目比后一个多1
				if (num < v1[i])//如果前一轮比较的得到的结果比后一轮比较得到的结果大，则保持前一轮的结果
					num = v1[i];
				v1[i] = num;//更新第i个小孩的糖果数目
			}
			else
			{
				num = v1[i];//如果没比后一个小孩的数字大，他得到的糖果数目维持第一轮的数目
			}
			sum += num;
		}
	}	
	return sum;
}
int main()
{
	int a[] = { 5,3,1 };
	vector<int> v(a, a + 3); 
	cout << candy(v) << endl;
	system("pause");
	return 0;
}

已通过LeetCode平台验证，其中一个实验结果如下：