[HEOI2014]大工程

一、题目

点此看题

二、解法

这道题设计三个子问题，我们逐个用$dp$解决。

• 代价和，考虑每条边的贡献，反应到点上，设$sum[u]$为子树$u$内关键点个之和，所以每条边的贡献为$sum[u]\times (k-sum[u])$（此处边权为1），然后贡献求即可。
• 最小路径，定义$dp[u][0/1]$为子树内到$u$的最短路$/$次短路，然后答案为$min(dp[u][0]+dp[u][1])$，转移类比树的直径，值得注意的是我们要把初值赋成$inf$，关键点的初值$dp[u][0]$赋为$0$。
• 最大路径，类比上面的解法就行了，初值赋为$-inf$，关键点的$dp[u][0]$赋为$0$。

由于$k$的和是一定的，直接套虚树板子就可以了，时间复杂度$O(n\log n)$。

#include <cstdio>
#include <iostream> 
#include <algorithm>
#include <stack>
using namespace std;
#define LL long long
#define inf 0x3f3f3f3f
const int MAXN = 1000005;
int read()
{
	int x=0,flag=1;char c;
	while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
	while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*flag;
}
int n,m,k,k1,k2,cnt,tot,f[MAXN],dfin[MAXN],dfou[MAXN],vis[MAXN],fa[MAXN][20];
int dep[MAXN],mi[MAXN][2],ma[MAXN][2],sum[MAXN],a[MAXN*2];LL ans;//最短，次短，最长，次长，点数，和 
stack<int> s;
struct edge
{
	int v,next;
}e[MAXN*2];
bool cmp(int a,int b)
{
	int t1=a>0?dfin[a]:dfou[-a];
	int t2=b>0?dfin[b]:dfou[-b];
	return t1<t2;
}
void dfs(int u,int p)
{
	dfin[u]=++cnt;
	dep[u]=dep[p]+1;
	fa[u][0]=p;
	for(int i=1;i<=19;i++)
		fa[u][i]=fa[fa[u][i-1]][i-1];
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==p) continue;
		dfs(v,u);
	}
	dfou[u]=++cnt;
}
int get(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=19;i>=0;i--)
		if(dep[fa[u][i]]>=dep[v])
			u=fa[u][i];
	if(u==v) return u;
	for(int i=19;i>=0;i--)
		if(fa[u][i]^fa[v][i])
			u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}
int main()
{
	n=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();
		e[++tot]=edge{v,f[u]},f[u]=tot;
		e[++tot]=edge{u,f[v]},f[v]=tot;
	}
	dfs(1,0);
	m=read();
	for(int i=1;i<=n;i++)
	{
		mi[i][0]=mi[i][1]=inf;
		ma[i][0]=ma[i][1]=-inf;
	}
	while(m--)
	{
		k=k1=k2=read();
		for(int i=1;i<=k;i++)
		{
			a[i]=read();vis[a[i]]=1;
			mi[a[i]][0]=ma[a[i]][0]=0;sum[a[i]]=1;
		}
		sort(a+1,a+1+k,cmp);
		for(int i=1;i<k1;i++)
		{
			int t=get(a[i],a[i+1]);
			if(!vis[t])
				a[++k]=t,vis[t]=1;
		}
		if(!vis[1]) a[++k]=1,vis[1]=1;
		k1=k;
		for(int i=1;i<=k1;i++)
			a[++k]=-a[i];
		sort(a+1,a+1+k,cmp);
		int Max=0,Min=inf;
		for(int i=1;i<=k;i++)
		{
			if(a[i]>0)
				s.push(a[i]);
			else
			{
				int t=s.top();s.pop();
				Max=max(Max,ma[t][0]+ma[t][1]);
				Min=min(Min,mi[t][0]+mi[t][1]);
				if(t==1) break; 
				int p=s.top(),dis=dep[t]-dep[p];
				sum[p]+=sum[t];
				ans+=1ll*sum[t]*(k2-sum[t])*dis;
				if(mi[p][0]>mi[t][0]+dis)
				{
					mi[p][1]=mi[p][0];
					mi[p][0]=mi[t][0]+dis;
				}
				else
					mi[p][1]=min(mi[p][1],mi[t][0]+dis);
				if(ma[p][0]<ma[t][0]+dis)
				{
					ma[p][1]=ma[p][0];
					ma[p][0]=ma[t][0]+dis;
				}
				else
					ma[p][1]=max(ma[p][1],ma[t][0]+dis);
				mi[t][0]=mi[t][1]=inf;
				ma[t][0]=ma[t][1]=-inf;sum[t]=0;
				vis[t]=0;//注意清访问标记 
			}
		}
		mi[1][0]=mi[1][1]=inf;
		ma[1][0]=ma[1][1]=-inf;sum[1]=0;
		vis[1]=0;
		printf("%lld %d %d\n",ans,Min,Max);
		ans=0;
	}
}