P4103 [HEOI2014]大工程【虚树】

洛谷

---

SOL

虚树板题一道。。。

个人觉得这题比 “消耗战”更适合练板子（树形dp更为简单）

说正解。

注意到$\sum |p|\le 2e6$，这提示我们建一颗虚树。

建好后，2和3问就是求一个树上最短路，最长路，不再赘述。

简单说一下1，我们对于每一条边统计有多少点对经过它。
具体的，树形dp的时候算出，即为$siz[son]\ast (tot-siz[son)$。

#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define ri register int
#define in red()
#define gc getchar()
#define ll long long
#define cs const
inline int red(){
	int num=0,f=1;char c=gc;
	for(;!isdigit(c);c=gc)if(c=='-')f=-1;
	for(;isdigit(c);c=gc)num=(num<<1)+(num<<3)+(c^48);
	return num*f;
}
cs int N=1e6+10;
cs ll inf=1e14;
int head[N],to[N<<1],nxt[N<<1],cnt=0,dep[N],n,m,dfn[N],tot;
inline void adde(int u,int v){
	nxt[++cnt]=head[u];head[u]=cnt;to[cnt]=v;
	nxt[++cnt]=head[v];head[v]=cnt;to[cnt]=u;
}
namespace SP{
	int siz[N],fa[N],top[N],son[N];
	inline void dfs1(int u){
		dep[u]=dep[fa[u]]+1;
		dfn[u]=++tot;
		siz[u]=1;
		for(ri i=head[u];i;i=nxt[i]){
			int v=to[i];
			if(v^fa[u]){
				fa[v]=u;
				dfs1(v);
				siz[u]+=siz[v];
				if(siz[v]>siz[son[u]])son[u]=v;
			}
		}
	}
	inline void ddfs1(int u,int f){
		top[u]=f;
		if(son[u])ddfs1(son[u],f);
		for(ri i=head[u];i;i=nxt[i]){
			int v=to[i];
			if(v^fa[u]&&v^son[u])ddfs1(v,v);
		}
	}
	inline int lca(int x,int y){
		while(top[x]^top[y]){
			if(dep[top[x]]<dep[top[y]])swap(x,y);
			x=fa[top[x]];
		}
		return dep[x]<dep[y] ? x : y;
	}
	inline void init(){
		dfs1(1);ddfs1(1,1);
	}
}
using SP::lca;
using SP::init;
inline bool cmp(cs int &a,cs int &b){return dfn[a]<dfn[b];}
typedef pair<int,int> pi;
#define fi first
#define se second
vector<pi> g[N];
inline void link(int x,int y){g[x].push_back(pi(y,dep[y]-dep[x]));}
ll maxn,siz[N],mx[N],mn[N],ans1,ans2,ans3;
bool vis[N];
//vis标记打上，清掉 
void dfs2(int u,int fa){
	if(vis[u])mx[u]=mn[u]=0,siz[u]=1;
	else mx[u]=-inf,mn[u]=inf,siz[u]=0;
	for(ri i=g[u].size()-1;i>=0;--i){
		int v=g[u][i].fi;
		if(v==fa)continue;
		dfs2(v,u);
		siz[u]+=siz[v];
		ans1+=siz[v]*(maxn-siz[v])*g[u][i].se;
		ans2=min(ans2,mn[u]+g[u][i].se+mn[v]);
		mn[u]=min(mn[u],g[u][i].se+mn[v]);
		ans3=max(ans3,mx[u]+g[u][i].se+mx[v]);
		mx[u]=max(mx[u],g[u][i].se+mx[v]);
	}
	g[u].clear();
}
inline void work(vector<int> &f){
	vector<int> stk;stk.clear();
	sort(f.begin(),f.end(),cmp);
	for(ri i=0,up=f.size();i<up;++i){
		if(!stk.size()){stk.push_back(f[i]);continue;}
		int L=lca(*stk.rbegin(),f[i]);
		while(stk.size()>1&&dfn[*(--(--stk.end()))]>=dfn[L])link(*(--(--stk.end())),*stk.rbegin()),stk.pop_back();
		if((*stk.rbegin())^L)link(L,*stk.rbegin()),stk.pop_back(),stk.push_back(L);
		stk.push_back(f[i]);
	}	
	while(stk.size()>1)link(*(--(--stk.end())),*stk.rbegin()),stk.pop_back();
	maxn=f.size();
	ans1=0;ans2=inf;ans3=-inf;
	dfs2(*stk.begin(),0);
	cout<<ans1<<' '<<ans2<<' '<<ans3<<'\n';
	
}
vector<int> cot;
signed main(){
//	freopen("data.in","r",stdin);
	n=in;
	for(ri i=1;i<n;++i)adde(in,in);
	init();
	m=in;
	while(m--){
		cot.clear();
		int k=in;
		for(ri i=1;i<=k;++i)cot.push_back(in),vis[cot[i-1]]=1;
		work(cot);
		for(ri i=1;i<=k;++i)vis[cot[i-1]]=0;
	}

	return 0;
}