BZOJ3611: [Heoi2014]大工程【虚树+树形DP】

3611: [Heoi2014]大工程

首先我们肯定会想到DP求这个答案，但是发现点数实在太多，然后建虚树就可以了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=1e6+5;
typedef long long LL;
struct Edge{
	int tot,lnk[MAXN],nxt[MAXN<<1],son[MAXN<<1],W[MAXN<<1];
	void Add(int x,int y,int w){nxt[++tot]=lnk[x];W[tot]=w;lnk[x]=tot;son[tot]=y;}
}E,S;
#include<cctype>
int read(){
	int ret=0;char ch=getchar();bool f=1;
	for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
	for(; isdigit(ch);ch=getchar()) ret=(ret<<1)+(ret<<3)+ch-48;
	return f?ret:-ret;
}
int n,m,q,tim,IN[MAXN],OUT[MAXN],Fa[MAXN][30],Dep[MAXN],a[MAXN],Stk[MAXN],Top;bool vis[MAXN],w[MAXN];
int F_MAX[MAXN],Ans_MAX,F_MIN[MAXN],Ans_MIN;LL F[MAXN],Siz[MAXN],Ans;
void DFS(int x,int fa){
	IN[x]=++tim;Fa[x][0]=fa,Dep[x]=Dep[fa]+1;
	for(int j=E.lnk[x];j;j=E.nxt[j]) if(E.son[j]!=fa) DFS(E.son[j],x);
	OUT[x]=++tim;
}
void INIT(){
	for(int j=1;(1<<j)<=n;j++)
	for(int i=1;i<=n;i++) Fa[i][j]=Fa[Fa[i][j-1]][j-1];
}
int LCA(int p,int q){
	if(Dep[p]<Dep[q]) swap(p,q);
	int Del=Dep[p]-Dep[q];
	for(int i=0;(1<<i)<=Del;i++) if((1<<i)&Del) p=Fa[p][i];
	if(p==q) return p;
	for(int i=log2(n);i>=0;i--)
	if(Fa[p][i]!=Fa[q][i]) p=Fa[p][i],q=Fa[q][i];
	return Fa[p][0];
}
bool cmp(int x,int y){return (x<0?OUT[-x]:IN[x])<(y<0?OUT[-y]:IN[y]);}
void DP(int x,int fa){
	if(w[x]) Siz[x]=1,F[x]=0,F_MAX[x]=0,F_MIN[x]=0;
	else Siz[x]=0,F[x]=0,F_MAX[x]=0,F_MIN[x]=1e9;
	LL Sum=0,Number=0;int MAX=0,MIN=1e9;
	for(int j=S.lnk[x];j;j=S.nxt[j])
	if(S.son[j]!=fa){
		DP(S.son[j],x);
		if(Number)
		Ans_MAX=max(F_MAX[S.son[j]]+MAX+S.W[j],Ans_MAX),
		Ans_MIN=min(F_MIN[S.son[j]]+MIN+S.W[j],Ans_MIN),
		Ans+=(F[S.son[j]]+Siz[S.son[j]]*S.W[j])*Number+Sum*Siz[S.son[j]];
		if(w[x])
		Ans+=F[S.son[j]]+S.W[j]*Siz[S.son[j]],
		Ans_MAX=max(F_MAX[S.son[j]]+S.W[j],Ans_MAX),
		Ans_MIN=min(F_MIN[S.son[j]]+S.W[j],Ans_MIN);
		Number+=Siz[S.son[j]],Sum+=F[S.son[j]]+Siz[S.son[j]]*S.W[j];MAX=max(F_MAX[S.son[j]]+S.W[j],MAX),MIN=min(F_MIN[S.son[j]]+S.W[j],MIN);
	}
	F[x]=Sum,F_MAX[x]=max(F_MAX[x],MAX),F_MIN[x]=min(F_MIN[x],MIN),Siz[x]+=Number;
}
int main(){
//	freopen("3611.in","r",stdin);
//	freopen("3611.out","w",stdout);
	n=read();
	for(int i=1,x,y;i<n;i++) x=read(),y=read(),E.Add(x,y,1),E.Add(y,x,1);
	q=read();DFS(1,0),INIT();
	while(q--){
		m=read();
		for(int i=1;i<=m;i++) a[i]=read(),vis[a[i]]=w[a[i]]=1;
		sort(a+1,a+1+m,cmp);
		for(int i=2;i<=m;i++){
			int fa=LCA(a[i],a[i-1]);
			if(!vis[fa]) vis[fa]=1,a[++m]=fa;
		}
		if(!vis[1]) vis[1]=1,a[++m]=1;
		for(int i=1;i<=m;i++) vis[a[i]]=0;//clear
		for(int i=1,END=m;i<=END;i++) a[++m]=-a[i];
		sort(a+1,a+1+m,cmp);
		for(int i=1;i<=m;i++)
		if(a[i]<0) Top--;else S.Add(Stk[Top],a[i],Dep[a[i]]-Dep[Stk[Top]]),S.Add(a[i],Stk[Top],Dep[a[i]]-Dep[Stk[Top]]),Stk[++Top]=a[i];
		Ans=0,Ans_MAX=0,Ans_MIN=1e9;
		DP(1,0);
		printf("%lld %d %d\n",Ans,Ans_MIN,Ans_MAX);
		for(int i=1;i<=m;i++) if(a[i]>0) w[a[i]]=0,S.lnk[a[i]]=0;S.tot=0;
	}
	return 0;
}