bzoj 2151 种树 —— 思路+链表

时间:2023-03-10 04:40:09
bzoj 2151 种树 —— 思路+链表

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2151

先都放进堆里取最大的,但选了一个就不能选它两边的,所以可能不是最优,要有“反悔”的措施;

可以取出一个后把它两边的位置 l,r 在链表中删除,然后再加入一个元素 a[x] = a[l] + a[r] - a[x],如果日后选了这个元素,就表示“反悔”了,当初不选 x 而是同时选了两边的(同时选比只选一个或不选更优);

而此时这个 x 的两边实际上是 l-1 和 r+1,因为其实选的是 l,r,这点用链表删除和查询就可以体现,更何况两边也可能被这样合并过;

再记一个 vis 表示这个位置从堆里取出时能不能选了,和链表的删除一致;

因为每次取出都会增加一个选择的位置,所以取出 m 次即可。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
int const xn=2e5+;
int n,m,a[xn],pr[xn],nt[xn],ans;
bool vis[xn];
struct N{
int v,id;
bool operator < (const N &y) const
{return v<y.v;}
};
priority_queue<N>q;
int rd()
{
int ret=,f=; char ch=getchar();
while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
return f?ret:-ret;
}
void del(int x){nt[pr[x]]=nt[x]; pr[nt[x]]=pr[x];}
int main()
{
n=rd(); m=rd();
for(int i=;i<=n;i++)
{
a[i]=rd();
q.push((N){a[i],i});
}
if(n<*m){puts("Error!"); return ;}
for(int i=;i<n;i++)pr[i]=i-,nt[i]=i+;
pr[]=n; nt[]=; pr[n]=n-; nt[n]=;
for(int i=;i<=m;i++)
{
int x=q.top().id,w=q.top().v; q.pop();
while(vis[x])x=q.top().id,w=q.top().v,q.pop();
int l=pr[x],r=nt[x]; ans+=w;
del(l); del(r); vis[l]=; vis[r]=; a[x]=a[l]+a[r]-w;
q.push((N){a[x],x});
}
printf("%d\n",ans);
return ;
}