洛谷 P1484 种树 题解

题面

这是一道标准的带反悔贪心；

利用大根堆来维护最大值：

当选择了num[i]后，反悔了，反之选择选了num[i-1]和num[i+1]时获利便增加了num[i-1]+num[i+1]-num[i]。

所以当num[i]被选时，我们就可以删去num[i-1]和num[i+1]，并把num[i]改成num[i-1]+num[i+1]-num[i]，放进堆中

#include <bits/stdc++.h>
#define int long long
using namespace std;
int a[];
priority_queue<pair<long long,long long> > q;
bool bo[];
int l[],r[];
signed main()
{
    int n,m;
    cin>>n>>m;
    for(register int i=;i<=n;i++){
        cin>>a[i];
        l[i]=i-;
        r[i]=i+;
        q.push(make_pair(a[i],i));
    }
    l[n+]=n;
    r[]=;
    long long ans=;
    for(int i=;i<=m;i++){
        while(bo[q.top().second]){
            q.pop();
        }
        int x=q.top().second;
        long long tmp=q.top().first;
        q.pop();
        if(tmp<) break;
        ans+=tmp;
        a[x]=a[l[x]]+a[r[x]]-a[x];
        bo[l[x]]=;
        bo[r[x]]=;
        l[x]=l[l[x]];
        r[x]=r[r[x]];
        r[l[x]]=x;
        l[r[x]]=x;
        q.push(make_pair(a[x],x));
    }
    cout<<ans;
}