分类讨论
比前者大【暂定此位置糖果为前者+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;
}
};