leetcode---candy---dp

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?

class Solution {
public:
    int candy(vector<int> &ratings) 
    {
        int n = ratings.size();
        vector<int> dp(n, 0);
        dp[0] = 1;
        int sum = 1;
        if(n == 0)
        {
            return 0;
        } 

        for(int i=1; i<n; i++)
        {
            if(ratings[i] > ratings[i-1])
            {
                dp[i] = dp[i-1] + 1;
                sum += dp[i];
            }
            else if(ratings[i] < ratings[i-1])
            {
                if(i == 1)
                {
                    dp[0] = 2;
                    dp[1] = 1;
                    sum = 3;
                }
                else
                {
                    dp[i] = 1;
                    int k = i-1;
                    if(ratings[k] <= ratings[k-1])
                    {
                        while(k > 0 && ratings[k] < ratings[k-1] && dp[k]+1 == dp[k-1])
                        {
                            dp[k] += 1;
                            k--;
                            sum++;
                        }
                        dp[k] += 1;
                        sum++;
                    }
                    sum += dp[i];
                }
            }
            else
            {
                dp[i] = 1;
                sum += dp[i];
            }
        }
        return sum;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值