LeetCode Algorithms 135. Candy 题解

来源:互联网 发布:c#图形编程 书籍 编辑:程序博客网 时间:2024/06/18 15:56

There are N children standing in a line. Each child is assigned a
rating value.

You are giving candies to these children subjected to the following requirements:
Each child must have at least one candy. Children with a higher rating get more candies than their neighbors. What is the minimum candies you must give?

题目意思是N个小孩一字排开,每个小孩有一个分数,分数比旁边的小孩高的要比旁边的小孩得到更多的糖果,每个小孩至少得到一个糖果,求最少需要多少糖果。

运用动态规划的思想,如果旁边的小孩分数比当前小孩要低,那么当前小孩的糖果数取当前小孩的糖果数和旁边小孩的糖果数+1的最大值,但是旁边的小孩有左右之分,为了减少一边的判断,先从左或者右开始初始化一次每个小孩的糖果数。

例如从队伍左开始初始化,判断左边小孩分数与当前小孩分数大小,如果当前小孩分数高,则当前小孩的糖果数是左边小孩糖果数+1,如此遍历一遍数组后,就能保证每个小孩的糖果数关系对于左边的小孩来说是正确的。

初始化之后可以使用简单的动态规划思想,从队伍右边开始判断,当前小孩的分数与右边小孩的分数,如果当前小孩分数较高,则取当前小孩的糖果数与右边小孩糖果数+1的最大值。这里要注意从右边开始判断,因为每次判断的是右边的分数较低者,要先确定小孩右边的糖果数最终是多少。

代码如下:

int Solution::candy(vector<int>& ratings) {    int nums = ratings.size();    if (nums <= 1) return nums;    vector<int> res(nums, 1);    for (int i = 1; i < nums; i++) {        if (ratings[i] > ratings[i - 1]) {            res[i] = res[i - 1] + 1;        }    }    for (int i = nums - 1; i > 0; i--) {        if (ratings[i] < ratings[i - 1]) {            res[i - 1] = max(res[i] + 1, res[i - 1]);        }    }    int min_res = 0;    for (int i = 0; i < nums; i++) {        min_res += res[i];    }    return min_res;}
0 0
原创粉丝点击