bzoj 3611(洛谷 4103) [Heoi2014]大工程——虚树

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3611

　　　https://www.luogu.org/problemnew/show/P4103

虚树的边权就是两端点的 dep 差。

求最短边可以像求最长边一样记录一个 d2[ cr ] 表示 cr 到其子树里最近的关键点的距离。

可以用 vis[ ] 记录哪些点是关键点。在 DP 的 dfs 里清空数组真是舒服。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
int rdn()
{
  int ret=0;bool fx=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();}
  while(ch>='0'&&ch<='9')ret=ret*10+ch-'0',ch=getchar();
  return fx?ret:-ret;
}
int Mx(int a,int b){return a>b?a:b;}
int Mn(int a,int b){return a<b?a:b;}
const int N=1e6+5,K=25;
int n,hd[N],xnt,to[N<<1],nxt[N<<1];
int dep[N],dfn[N],tim,bin[K],pre[N][K];
bool cmp(int x,int y){return dfn[x]<dfn[y];}
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
void dfs(int cr,int fa)
{
  dfn[cr]=++tim;
  dep[cr]=dep[fa]+1; pre[cr][0]=fa;
  for(int t=1;bin[t]<=dep[cr];t++)
    pre[cr][t]=pre[pre[cr][t-1]][t-1];
  for(int i=hd[cr],v;i;i=nxt[i])
    if((v=to[i])!=fa) dfs(v,cr);
}
int get_lca(int x,int y)
{
  if(dep[x]<dep[y])swap(x,y);
  int d=dep[x]-dep[y];
  for(int t=0;bin[t]<=d;t++)
    if(d&bin[t])x=pre[x][t];
  if(x==y)return x;
  for(int t=20;t>=0;t--)
    if(pre[x][t]!=pre[y][t])
      x=pre[x][t],y=pre[y][t];
  return pre[x][0];
}
namespace Tr{
  int hd[N],xnt,to[N],nxt[N],len,mn; ll sm;
  int p[N],tot,sta[N],top,siz[N],dis[N],d2[N];
  bool vis[N];
  void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
  void dfs(int cr)
  {
    dis[cr]=0; siz[cr]=vis[cr]; d2[cr]=(vis[cr]?0:N);
    for(int i=hd[cr],v;i;i=nxt[i])
      {
    dfs(v=to[i]); int w=dep[v]-dep[cr];
    sm+=(ll)w*siz[v]*(tot-siz[v]);
    len=Mx(len,dis[cr]+dis[v]+w); dis[cr]=Mx(dis[cr],dis[v]+w);
    mn=Mn(mn,d2[cr]+d2[v]+w); d2[cr]=Mn(d2[cr],d2[v]+w);
    siz[cr]+=siz[v];
      }
    vis[cr]=0; hd[cr]=0;
  }
  void solve()
  {
    tot=rdn(); xnt=0;//xnt
    for(int i=1;i<=tot;i++)p[i]=rdn(),vis[p[i]]=1;
    sort(p+1,p+tot+1,cmp); sta[top=1]=p[1];//cmp!!!
    for(int i=2;i<=tot;i++)
      {
    int u=p[i], lca=get_lca(u,sta[top]);
    while(top&&dfn[lca]<dfn[sta[top]])
      {
        if(dfn[sta[top-1]]<dfn[lca])
          add(lca,sta[top]);
        else add(sta[top-1],sta[top]);
        top--;
      }
    if(sta[top]!=lca)sta[++top]=lca;
    sta[++top]=u;
      }
    for(int i=1;i<top;i++)add(sta[i],sta[i+1]);
    sm=0; mn=N; len=0;
    dfs(sta[1]); printf("%lld %d %d
",sm,mn,len);
  }
}
int main()
{
  bin[0]=1;for(int i=1;i<=21;i++)bin[i]=bin[i-1]<<1;
  n=rdn();
  for(int i=1,u,v;i<n;i++)
    u=rdn(),v=rdn(),add(u,v),add(v,u);
  dfs(1,0); int Q=rdn();
  while(Q--)Tr::solve();
  return 0;
}