luogu1484 种树 (优先队列)

我每次都想选那个最大的、或者是它旁边的两个一起选，如果这两个一起选会大于那个最大的的话

那我就先把那个最大的选了，再提供一个反悔的选项（类似于网络流的思路？），就是我可以再把种的树换成它旁边那两个，也是相当于又多种了一个，这个的权值是v[l]+v[r]-v[x]

所以用一个双向链表维护这个l、r，用一个优先队列维护这个最大值就好了

 #include<bits/stdc++.h>
 #define pa pair<ll,int>
 #define CLR(a,x) memset(a,x,sizeof(a))
 using namespace std;
 typedef long long ll;
 const int maxn=5e5+;

 inline ll rd(){
     ll x=;char c=getchar();int neg=;
     while(c<''||c>''){if(c=='-') neg=-;c=getchar();}
     while(c>=''&&c<='') x=x*+c-'',c=getchar();
     return x*neg;
 }

 priority_queue<pa> q;
 int N,K,nxt[maxn],pre[maxn];
 ll v[maxn];
 bool flag[maxn];

 int main(){
     //freopen("","r",stdin);
     int i;
     N=rd(),K=rd();
     for(i=;i<=N;i++)
         v[i]=rd(),q.push(make_pair(v[i],i));
     for(i=;i<=N;i++)
         nxt[i]=i+,pre[i]=i-;
     nxt[]=,pre[N+]=N;
     ll ans=;
     while(K&&!q.empty()){
         pa p=q.top();q.pop();
         int i=p.second,a=pre[i],b=nxt[i];
         if(flag[i]) continue;
         if(p.first<=) break;
         ans+=p.first;K--;
         flag[a]=flag[b]=;
         v[i]=v[a]+v[b]-p.first;
         pre[i]=pre[a],nxt[pre[a]]=i;
         nxt[i]=nxt[b],pre[nxt[b]]=i;
         q.push(make_pair(v[i],i));
     }
     printf("%lld\n",ans);
     return ;
 }