HEOI 2014 大工程 题解

题目传送门

题目大意： 给出一棵树，边权为 $1$，每次询问给出若干个关键点，问这些关键点两两之间的最长的路径、最短的路径、路径长度之和。

题解

造出虚树然后一波 $dp$ 就做完了。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 1000010
#define inf 999999999
#define ll long long

int n,m,q,a[maxn];
struct edge{int x,y,z,next;}e[maxn<<1];
int first[maxn],len=0;
void buildroad(int x,int y,int z=0){e[++len]=(edge){x,y,z,first[x]};first[x]=len;}
int deep[maxn],f[maxn][20],id[maxn],tot=0;
void dfs(int x,int fa,int dep)
{
	f[x][0]=fa;deep[x]=dep;id[x]=++tot;
	for(int i=first[x];i;i=e[i].next)
	if(e[i].y!=fa)dfs(e[i].y,x,dep+1);
}
int zhan[maxn],t;
int get_lca(int x,int y)
{
	if(deep[x]>deep[y])swap(x,y);
	for(int i=19;i>=0;i--)if(deep[f[y][i]]>=deep[x])y=f[y][i];
	if(x!=y){for(int i=19;i>=0;i--)if(f[x][i]!=f[y][i])x=f[x][i],y=f[y][i];x=f[x][0];}
	return x;
}
void print();
void add(int x)
{
	if(x==1)return; if(t==1)return (void)(zhan[++t]=x);
	int lca=get_lca(x,zhan[t]);if(lca==zhan[t])return (void)(zhan[++t]=x);
	while(deep[zhan[t-1]]>=deep[lca])buildroad(zhan[t-1],zhan[t],deep[zhan[t]]-deep[zhan[t-1]]),t--;
	if(zhan[t]!=lca)buildroad(lca,zhan[t],deep[zhan[t]]-deep[lca]),zhan[t]=lca;zhan[++t]=x;
}
bool v[maxn];
//mn,mx分别记录子树内关键点延伸上来的最长路径和最短路径，ma，mi记录答案
int mi,ma,mx[maxn],mn[maxn],size[maxn];//size记录子树内关键点数量 
ll g[maxn],sum[maxn];//g记录所有关键点延伸上来的路径长度和，sum记录子树内所有路径之和 
int dp(int x)
{
	size[x]=v[x];sum[x]=g[x]=0;
	mx[x]=-inf,mn[x]=inf;
	for(int i=first[x];i;i=e[i].next)
	{
		int y=e[i].y;dp(y);
		ma=max(ma,mx[y]+e[i].z+mx[x]);
		mi=min(mi,mn[y]+e[i].z+mn[x]);
		mx[x]=max(mx[x],mx[y]+e[i].z);
		mn[x]=min(mn[x],mn[y]+e[i].z);
		sum[x]+=g[x]*size[y]+(g[y]+e[i].z*size[y])*size[x]+sum[y];
		size[x]+=size[y];
		g[x]+=g[y]+e[i].z*size[y];
	}
	if(v[x])mi=min(mi,mn[x]),ma=max(ma,mx[x]),mn[x]=0,mx[x]=max(mx[x],0);
	v[x]=false;
}
bool cmp(int x,int y){return id[x]<id[y];}

int main()
{
	scanf("%d",&n);
	for(int i=1,x,y;i<n;i++)scanf("%d %d",&x,&y),
	buildroad(x,y),buildroad(y,x); dfs(1,0,1);
	for(int j=1;j<=19;j++)for(int i=1;i<=n;i++)f[i][j]=f[f[i][j-1]][j-1];
	scanf("%d",&q);while(q--)
	{
		while(len)first[e[len].x]=0,len--;
		scanf("%d",&m);zhan[t=1]=1;
		for(int i=1;i<=m;i++)scanf("%d",&a[i]),v[a[i]]=true;
		sort(a+1,a+m+1,cmp);for(int i=1;i<=m;i++)add(a[i]);
		while(t)buildroad(zhan[t-1],zhan[t],deep[zhan[t]]-deep[zhan[t-1]]),t--;
		mi=inf;ma=0;dp(1);printf("%lld %d %d\n",sum[1],mi,ma);
	}
}