[leetcode] Candy

最新推荐文章于 2023-02-28 14:31:32 发布

原创最新推荐文章于 2023-02-28 14:31:32 发布 · 353 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

222 篇文章

订阅专栏

from : https://leetcode.com/problems/candy/

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:

思路：

举例，

(1) 4 5 5 8 9 3 2 6 7

(2) 1 2 1 2 3 1

(3) 2 1

(4) 4 3 2 1

(5) 5 4 3 2 1

上面是遍历的过程。

（a）如果前一个比后一个大，那么后一个的糖就比前一个多一块；

（b）如果前一个和后一个一样大，那么后一个就分得一块；

（c）如果前一个比后一个小，那么

（c.1）如果前一个比1大，那么这个就分一块；这个必然是第一次分得1块的，那么其前面一个人（如果存在）可能比当前的人多一块以上，记这个人为P，见(2)；

（c.2）如果前一个等于1，那么表示这个不能更小了，就要给前面的多分一块；如果P比后面的人多一块以上，那么只要给这个人后面的每个人都多分一块，见(3)；

（c.3）如果不满足上面的两点，那么所有P和P后面的人都要多分一块糖，见(4)，见(5)。

public class Solution {
    public int candy(int[] ratings) {
        if(null == ratings || 0 == ratings.length) {
            return 0;
        }
        int n = 1;
        int high = 1, low = 0, prenum = 1, hidx = 0;
        for(int i=1,len = ratings.length; i<len; ++i) {
            int cur = ratings[i], pre = ratings[i-1];
            if(cur >= pre) {
                if(cur == pre) {
                    prenum = 1;
                    n += 1;
                } else {
                    n += ++prenum;
                }
                hidx = i;
                high = prenum;
            } else {
                if(1 < prenum) {
                    n += 1;
                    prenum = 1;
                    low = 1;
                } else if(low+1 < high) {
                    n += ++low;
                } else {
                    ++low;
                    n += ++high;
                }
            }
        }
        return n;
    }
}

class Solution {
public:
    int candy(vector<int>& ratings) {
        int n = 1;
        int high = 1, low = 0, prenum = 1, len = ratings.size();
        
        for(int i=1, len = ratings.size(); i<len; ++i) {
            int cur = ratings[i], pre = ratings[i-1];
            if(cur >= pre) {
                if(cur == pre) {
                    prenum = 1;
                    n += 1;
                } else {
                    n += ++prenum;
                }
                high = prenum;
            } else {
                if(1 < prenum) {
                    ++n;
                    prenum = low = 1;
                } else if(low+1 < high) {
                    n += ++low;
                } else {
                    ++low;
                    n += ++high;
                }
            }
        }
        
        return n*(len > 0);
    }
};