leetcode 135. 分发糖果 (相邻的孩子中,评分高的必须糖果更多) 思维

在这里插入图片描述

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;
    }
};
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页