1. 程式人生 > >BZOJ3252: 攻略 可並堆

BZOJ3252: 攻略 可並堆

merge 可並堆 網上 -- 每一個 tdi printf swa main

網上有很多人說用dfs序+線段樹做...其實stl的堆可以...可並堆可以...很多奇奇怪怪的東西都能做...

可並堆比較好想...也比較好寫...

分析:

首先,這是一個網絡流做不了的題...數據太大...

其次...我們可以這樣考慮一下,這個點的子樹中,將這個點的權值僅更新給最大的那個就能滿足

之後,在每一個葉子節點上,建立一個大根堆,dfs一遍,將子節點的堆合並,之後找到根節點,將根節點的權值加上當前位置的價值

最後,根節點中前k大的權值和即為答案...

附上代碼,精簡可行

#include <cstdio>
#include <algorithm>
#include <queue>
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;
#define N 200005
#define ll long long
struct node
{
	int ls,rs,dis;
	ll x;
}mp[N<<1];
struct no
{
	int to,next;
}e[N<<1];
int head[N],cnt,fa[N],a[N],n,K;
void add(int x,int y)
{
	e[cnt].to=y;
	e[cnt].next=head[x];
	head[x]=cnt++;
	return ;
}
int merge(int x,int y)
{
	if(!x)return y;
	if(!y)return x;
	if(mp[x].x<mp[y].x)swap(x,y);
	mp[x].rs=merge(mp[x].rs,y);
	if(mp[mp[x].rs].dis>mp[mp[x].ls].dis)swap(mp[x].ls,mp[x].rs);
	mp[x].dis=mp[mp[x].rs].dis+1;
	return x;
}
void dfs(int x,int from)
{
	for(int i=head[x];i!=-1;i=e[i].next)
	{
		int to1=e[i].to;
		if(to1!=from)
		{
			dfs(to1,x);
			fa[x]=merge(fa[to1],fa[x]);
		}
	}
	mp[fa[x]].x+=a[x];
}
int in1[N];
int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&K);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		in1[x]++,in1[y]++;
		add(x,y);
		add(y,x);
	}
	int rot=1,num=0;
	for(int i=1;i<=n;i++)
	{
		if(in1[1]==1&&in1[i]!=1)
		{
			rot=i;
		}else
		{
			num++;
			fa[i]=i;
		}
	}
	K=min(num,K);
	dfs(rot,0);
	ll ans=0;
	while(K--)
	{
		ans+=mp[fa[rot]].x;
		mp[fa[rot]].x=0;
		fa[rot]=merge(mp[fa[rot]].ls,mp[fa[rot]].rs);
	}
	printf("%lld\n",ans);
	return 0;
}

  

BZOJ3252: 攻略 可並堆