【BZOJ 2151】種樹(連結串列+貪心)
阿新 • • 發佈:2018-12-30
有一種顯然錯誤的做法:每次從堆裡取最大的,更新連結串列
比如 19,20,19 這樣的話會先選20。但是很顯然,選兩個19帶來的貢獻可能是更好的
這時我們就要想一種能做到“反悔”的方法
其實只需要做一點點修改,每次再push一個“19+19-20”進去,一樣是對的
那這個“19+19-20”的放在哪兒呢(編號)?只需要放在20的位置就行了,這樣是正確的
其實這樣做的本質 是重寫了這個問題 將其變成“N-2個物品中選M-1”個的形式,如此一直化歸,直到最後變成“N-2(M-1)個物品中選1個”
#include<bits/stdc++.h> #define N 200005 using namespace std; template<class T> inline void read(T &x) { x=0; int f=1; static char ch=getchar(); while((ch<'0'||ch>'9')&&ch!='-') ch=getchar(); if(ch=='-') f=-1,ch=getchar(); while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar(); x*=f; } priority_queue<pair<int,int> >q; int n,m,a[N],l[N],r[N],ans; bool vis[N]; int main() { read(n),read(m); if(m>(n/2)){ puts("Error!"); return 0; } for(int i=1;i<=n;i++) { read(a[i]); q.push(make_pair(a[i],i)); l[i]=i-1,r[i]=i+1; } l[1]=n,r[n]=1; while(m--) { while(vis[q.top().second]) q.pop(); int val=q.top().first; int now=q.top().second; q.pop(); ans+=val, a[now]=a[l[now]]+a[r[now]]-val; vis[l[now]]=vis[r[now]]=true; l[now]=l[l[now]],r[l[now]]=now; r[now]=r[r[now]],l[r[now]]=now; q.push(make_pair(a[now],now)); } cout<<ans; return 0; }