【LeetCode】分发糖果

【问题】老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

你需要按照以下要求，帮助老师给这些孩子分发糖果：

每个孩子至少分配到 1 个糖果。
相邻的孩子中，评分高的孩子必须获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？

示例 :

输入: [,,]

输出:

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

示例 :

输入: [,,]

输出:

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

第三个孩子只得到  颗糖果，这已满足上述两个条件。

解题思路：

【思路】首先我们对整个数组进行初始化为1，dp[n]中元素都为1，第一次从左向右进行遍历，如果右边的得分大于左边的得分，则右边分得的糖果数比坐边多一个，不能贪多哟！但是我们并没有更新左边得分大于右边得分的情况，因此需要第二次遍历！

第二次遍历从右向左，类似于第一次，当ratings[i]>ratings[i+1]的同时，还要满足dp[i]<=dp[i+1]，为什么要多一个条件呢？这是因为第一次遍历确定了部分糖果数的正确大小关系，第二次遍历不能更改，加上这个条件后，我们只更新第一次遍历错误的糖果数！

C++代码：时间复杂度O(2*n)=O(n)

class Solution {

public:

    int candy(vector<int>& ratings) {

        int n = ratings.size();

        if(n < )

            return n;

        vector<int> dp(n, );

        for(int i = ;i < n; i++){

            if(ratings[i-] < ratings[i]){

                dp[i] = dp[i-] + ;

            }

        }

        for(int i = n-; i >= ; i--){

            if(ratings[i] > ratings[i+] && dp[i] <= dp[i+]){

                dp[i] = dp[i+] + ;

            }

        }

        int sum = ;

        for(int i: dp){

            sum += i;

        }

        return sum;

    }

};

巴特西

【LeetCode】分发糖果

最新文章

热门文章