题目大意: 给出一棵树,边权为 1 1 1,每次询问给出若干个关键点,问这些关键点两两之间的最长的路径、最短的路径、路径长度之和。
题解
造出虚树然后一波 d p dp 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);
}
}