LeetCode OJ Candy

来源:互联网 发布:游戏中常用算法 编辑:程序博客网 时间:2024/05/17 07:01

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?

刚开始还以为很简单- -。

后来做了好久。

后来看到了大神的好简单代码- -。

https://oj.leetcode.com/discuss/10046/share-my-short-solution

class Solution {public:    int candy(vector<int> &ratings) {        int ans = 0;        int * lc = new int[ratings.size()];        int * rc = new int[ratings.size()];        for (int i = 0; i < ratings.size(); i++) lc[i] = rc[i] = 1;        for (int i = 1; i < ratings.size(); i++) if (ratings[i - 1] < ratings[i]) lc[i] = lc[i - 1] + 1;        for (int i = ratings.size() - 2; i >= 0; i--) if (ratings[i] > ratings[i + 1]) rc[i] = rc[i + 1] + 1;        for (int i = 0; i < ratings.size(); i++) ans += max(lc[i], rc[i]);        return ans;    }};


0 0
原创粉丝点击