BZOJ 3611: [Heoi2014]大工程虚树树形dp

最新推荐文章于 2019-12-05 20:21:09 发布

BlackJack_

最新推荐文章于 2019-12-05 20:21:09 发布

阅读量363

点赞数 1

分类专栏：虚树 —————————树树形dp —————————dp

本文链接：https://blog.csdn.net/BlackJack_/article/details/79300664

版权

—————————dp 同时被 3 个专栏收录

100 篇文章 0 订阅

订阅专栏

—————————树

53 篇文章 0 订阅

订阅专栏

树形dp

18 篇文章 0 订阅

订阅专栏

3611: [Heoi2014]大工程

Time Limit: 60 Sec Memory Limit: 512 MB
Submit: 1927 Solved: 797
[Submit][Status][Discuss]

Description

国家有一个大工程，要给一个非常大的交通网络里建一些新的通道。

我们这个国家位置非常特殊，可以看成是一个单位边权的树，城市位于顶点上。

在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径。

现在国家有很多个计划，每个计划都是这样，我们选中了 k 个点，然后在它们两两之间新建 C(k,2)条新通道。

现在对于每个计划，我们想知道：

1.这些新通道的代价和

2.这些新通道中代价最小的是多少

3.这些新通道中代价最大的是多少

Input

第一行 n 表示点数。

接下来 n-1 行，每行两个数 a,b 表示 a 和 b 之间有一条边。

点从 1 开始标号。接下来一行 q 表示计划数。

对每个计划有 2 行，第一行 k 表示这个计划选中了几个点。

第二行用空格隔开的 k 个互不相同的数表示选了哪 k 个点。

Output

输出 q 行，每行三个数分别表示代价和，最小代价，最大代价。

Sample Input

10
2 1
3 2
4 1
5 2
6 4
7 5
8 6
9 7
10 9
5
2
5 4
2
10 4
2
5 2
2
6 1
2
6 1

Sample Output

3 3 3
6 6 6
1 1 1
2 2 2
2 2 2

HINT

n<=1000000

q<=50000并且保证所有k之和<=2*n

一眼虚树题

写了仨dp 写完看看别的题解感觉自己是个土鳖...

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
void print_int(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print_int(x/10);putchar(x%10+'0');}
void print_ll(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print_ll(x/10);putchar(x%10+'0');}

const int N=1001000,inf=0X3f3f3f3f;

int last[N],ecnt;
struct EDGE{int to,nt;}e[N<<1];
inline void add(int u,int v)
{if(u==v) return ;e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}

int n,K;

int dfn[N],tim;
int dep[N],anc[N][21];

void init(int u)
{
	dfn[u]=++tim;
	for(int i=1;(1<<i)<=dep[u];++i)
		anc[u][i]=anc[anc[u][i-1]][i-1];
	for(int i=last[u],v;i;i=e[i].nt)
		if((v=e[i].to)!=anc[u][0])
		{
			dep[v]=dep[u]+1;
			anc[v][0]=u;
			init(v);
		}
}

inline int getlca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	int len=dep[u]-dep[v];
	for(int i=0;(1<<i)<=len;++i)
		if((1<<i)&len)
			u=anc[u][i];
	if(u==v) return u;
	for(int i=20;~i;--i)
		if(anc[u][i]!=anc[v][i])
			u=anc[u][i],v=anc[v][i];
	return anc[u][0];
}

ll g[N];
int book[N];
int size[N],f[N][2],ans[2];

void dp_1(int u)
{
	g[u]=0;
	size[u]=book[u];
	for(int i=last[u],v;i;i=e[i].nt)
	{
		v=e[i].to;dp_1(v);
		g[u]+=g[v]+1ll*size[v]*(dep[v]-dep[u]);
		size[u]+=size[v];
	}
}

void dp_2(int u)
{
	for(int i=last[u],v;i;i=e[i].nt)
	{
		v=e[i].to;
		g[v]=g[u]+1ll*(dep[v]-dep[u])*(K-(size[v]<<1));
		dp_2(v);
	}
}

void dp_3(int u)
{
	f[u][0]= book[u] ? dep[u] : inf;
	for(int i=last[u],v,fir=inf,sec=inf;i;i=e[i].nt)
	{
		v=e[i].to;dp_3(v);
		f[u][0]=min(f[u][0],f[v][0]);
		if(f[v][0]<=fir) sec=fir,fir=f[v][0];
		else if(f[v][0]<=sec) sec=f[v][0];
		book[u] ? ans[0]=min(ans[0],f[v][0]-dep[u]) : ans[0]=min(ans[0],fir+sec-(dep[u]<<1));
	}
}

void dp_4(int u)
{
	f[u][1]= book[u] ? dep[u] : 0;
	for(int i=last[u],v,fir=-inf,sec=-inf;i;i=e[i].nt)
	{
		v=e[i].to;dp_4(v);
		f[u][1]=max(f[u][1],f[v][1]);
		if(f[v][1]>=fir) sec=fir,fir=f[v][1];
		else if(f[v][1]>=sec) sec=f[v][1];
		if(book[u]) ans[1]=max(ans[1],f[v][1]-dep[u]);
		ans[1]=max(ans[1],fir+sec-(dep[u]<<1));
	}
	last[u]=0;
}

int p[N];

inline bool cmp(int x,int y)
{return dfn[x]<dfn[y];}

int st[N];

void solve()
{
	ecnt=0;
	K=read();
	for(int i=1;i<=K;++i) p[i]=read(),book[p[i]]=1;
	sort(p+1,p+1+K,cmp);
	int top(1);
	st[top]=1;
	for(int i=1,grd;i<=K;++i)
	{
		grd=getlca(p[i],st[top]);
		while(1)
		{
			if(dep[st[top-1]]<=dep[grd])
			{
				add(grd,st[top]);top--;
				if(st[top]!=grd) st[++top]=grd;
				break;
			}
			add(st[top-1],st[top]);top--;
		}
		if(st[top]!=p[i]) st[++top]=p[i];
	}
	while(--top) add(st[top],st[top+1]);
	ll res(0);ans[0]=inf;ans[1]=0;
	dp_1(1);dp_2(1);dp_3(1);dp_4(1);
	for(int i=1;i<=K;++i) res+=g[p[i]],book[p[i]]=0;
	print_ll(res>>1);putchar(' ');
	print_int(ans[0]);putchar(' ');
	print_int(ans[1]);putchar('\n');
}

int main()
{
	int n=read();
	register int i,u,v;
	for(i=1;i<n;++i)
		u=read(),v=read(),add(u,v),add(v,u);
	init(1);
	memset(last,0,sizeof(last));
	int Q=read();
	while(Q--) solve();
	return 0;
}