Leetcode 135. 分发糖果

分类讨论
比前者大【暂定此位置糖果为前者+1,这个糖果可能因为后面孩子太少而增多,不过前面的不受影响n=1】

比前者相等【暂定此位置糖果为1,这个糖果可能因为后面孩子太少而增多,不过前面的不受影响n=1】

比前者小
若前者等于1,说明此位置孩子糖果太少了,要将前面受影响位置都+1,并暂定此位置糖果为1,且影响位置的个数也+1【3 2 1->4 3 2 1】
若前者等于2,暂定此位置糖果为1,且影响位置的个数也+1【4 3 2->4 3 2 1】
若前者大于2,则暂定此位置糖果为1,且记录上一个孩子糖果个数【3->(3) 1,当1受影响变成2时,需要将3连起来(3) 1->(3) 2 1【3 2 1】】

class Solution {
public:
    int candy(vector<int>& ratings) {
        if (ratings.empty()) return 0;
        int ans = 1, t = 1, le = 0, n = 1;
        for (int i = 1; i < ratings.size(); ++i)
            if (ratings[i] > ratings[i - 1])
                n = 1, ans += ++t, le = 0;
            else if (ratings[i] == ratings[i - 1])
                le = 0, t = n = 1, ++ans;
            else if (t == 2)
                ++n, ans += --t;
            else if (t>2)
                le = t - 2, t = n = 1, ++ans;
            else {
                ans += ++n;
                if (le && !--le)
                    ++n;
            }
            return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/bendaai/article/details/80992798