LeetCode(135)—分发糖果

题目要求
N个孩子站成一排,每个人被分配了一个排名值。

你需要给孩子分配糖果:

  • 每个孩子至少有一个糖果
  • 较高排名的比左边较低排名的多,比右边较低排名的多。
    解题思路贪心算法
  • 从前往后遍历ratings数组,如果i+1个孩子的排名高于第i个孩子,则第i+1个孩子的糖果比第i个孩子的糖果多1.
  • 从后往前遍历ratings数组,如果i+1个孩子的排名高于第i个孩子,而且第i+1个孩子的糖果数<=第i个孩子的糖果数,则第i+1个孩子的糖果要调整位第i个孩子的糖果数+1。
#include<iostream>
#include<vector>

using namespace std;


int cand(vector<int>& ratings)
{
   //首先至少保证每个孩子拥有一个糖果
	int len = ratings.size();
	vector<int>  result(len, 1);
	
	//从左到右依次遍历,
	//如果i+1个孩子的排名高于第i个孩子,则第i+1个孩子的糖果比第i个孩子的糖果多1
	for (int i = 1; i < len; ++i)
	{
		if (ratings[i] > ratings[i-1])
		{
			result[i] = result[i-1] + 1;
		}
	}

	//从右向左遍历,依据上述规则
	for (int i = len - 2; i >= 0; --i)
	{
		if (ratings[i] > ratings[i+1] && result[i] <= result[i+1])
		{
			result[i] = result[i+1] + 1;
		}
	}
    
	return  accumulate(result.begin(), result.end(), 0);
	//int sum = 0;
	//for (int i = 0; i < result.size(); ++i)
	//{
	//	sum += result[i];
	//}
	//return sum;
}

int main()
{
	vector<int >  nums;
	nums.push_back(1);
	nums.push_back(2);
	nums.push_back(6);
	nums.push_back(5);
	nums.push_back(4);
	nums.push_back(3);

	printf("%d\n", cand(nums));
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43313035/article/details/89407063