bzoj2151: 種樹(貪心+堆)
阿新 • • 發佈:2019-02-05
題目傳送門
。
解法:
今天才學這種經典做法。。
如果選最大的然後刪除兩邊很顯然這種策略是錯誤的。
因為有可能兩邊加起來更優。
那麼上面的做法是無法反悔的。。
給他一個反悔的機會。
那麼就是:
假設選了i,那麼刪除i前後的兩個點。
然後把i這個點的權值變為a[前]+a[後]-a[i]。
這樣再選i的話就表示反悔了選了前後兩個。
維護一下這個位置是否刪除。這個位置的前一個未被刪除的是哪個。後一個未被刪除的是哪個就行了。
具體看程式碼。
程式碼實現:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<queue>
#include<iostream>
using namespace std;
struct node {
int w,c;node() {w=c=0;}
friend bool operator <(node n1,node n2){return n1.c<n2.c;}
};priority_queue<node> a;
int q[210000],h[210000],n,m,s[210000];bool v[210000];
int main() {
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {scanf("%d",&s[i]);q[i]=i-1;h[i]=i+1;node ss;ss.c=s[i];ss.w=i;a.push(ss);}
if(m*2>n){printf("Error!\n");return 0;}
q[1]=n;h[n]=1;memset(v,false,sizeof(v));int ans=0;
for(int i=1;i<=m;i++) {
while(v[a.top().w]==true)a.pop ();
node t=a.top();a.pop();ans+=t.c;int l=q[t.w],r=h[t.w];
node p;p.w=t.w;v[l]=true;p.c+=s[l];v[r]=true;p.c+=s[r];p.c-=t.c;
s[t.w]=p.c;q[t.w]=q[l];h[q[l]]=t.w;h[t.w]=h[r];q[h[r]]=t.w;a.push(p);
}printf("%d\n",ans);
return 0;
}