Leetcode 动态规划 Candy

来源:互联网 发布:国产好的指纹锁 知乎 编辑:程序博客网 时间:2024/04/29 17:52

本文为senlie原创,转载请保留此地址:http://blog.csdn.net/zhengsenlie


Candy

 Total Accepted: 16494 Total Submissions: 87468My Submissions

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)每个小孩至少一颗糖
2)评分高的小孩发的糖比他旁边两个小孩的多
思路:左右 dp
用一个数组 candy[n],表示第 i 个小孩所应该发的最少糖果数
数组 ratings[n] 表示每个小孩的评分
1.从左到右扫描一遍, candy[i] = 1, if ratings[i] <= ratings[i-1] ; candy[i] = candy[i-1] + 1, if ratings[i] > ratings[i-1]
2.从右到左扫描一遍, candy[i] = candy[i], if ratings[i] <= ratings[i+1] ; candy[i] = max(candy[i], candy[i+1] + 1), if ratings[i] > ratings[i+1]
3.accumulate(candy, candy + n, 0)

复杂度: 时间 O(n), 空间 O(n)

int candy(vector<int> &ratings){int n = ratings.size();vector<int> candy(n, 1);for(int i = 1; i < n; ++i){candy[i] = ratings[i] <= ratings[i - 1] ? 1 : candy[i - 1] + 1;}for(int i = n - 2; i > -1;--i){candy[i] = ratings[i] <= ratings[i + 1] ? candy[i] : max(candy[i], candy[i + 1] + 1);}return accumulate(candy.begin(), candy.end(), 0);}


0 0
原创粉丝点击