class Solution {
public:
int candy(vector<int>& a) {
int n=a.size(),ans=1,pre=1;
if(n<=1) return n;
for(int i=0;i<n-1;){//左开右闭
if(a[i+1]>a[i]){
int cnt=1;//上坡的左端点一定是1
while(i+1<n&&a[i+1]>a[i]) ans+=++cnt,++i;
pre=cnt;
continue;
}
if(a[i+1]<a[i]){//下坡前面可能是上坡,左端点可能不为1
int cnt=1,prei=i;//我们假设为1
while(i+1<n&&a[i+1]<a[i]) ans+=--cnt,++i;
if(cnt<1){
ans+=(i-prei)*(1-cnt);//[prei+1,i]都增加1-cnt个🍬
if(pre<1+1-cnt) ans+=1+1-cnt-pre;//也试图让prei(左端点)变成1+1-cnt,但其实左端点更大
}
continue;
}
while(i+1<n&&a[i+1]==a[i]) ans+=1,++i;
pre=1;
}
return ans;
}
};
leetcode 135. 分发糖果 (相邻的孩子中,评分高的必须糖果更多) 思维
最新推荐文章于 2022-04-25 22:45:55 发布