[虚树] BZOJ3611: [Heoi2014]大工程

虚树裸题

#include <cstdio>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N=1000010;

int n,m,cnt,t,G[N],l[N],r[N],dpt[N];
struct edge{
  int t,nx;
}E[N<<1];

inline char nc(){
  static char buf[100000],*p1=buf,*p2=buf;
  return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &x){
  char c=nc(); x=0;
  for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}

inline void addedge(int x,int y){
  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;
  E[++cnt].t=x; E[cnt].nx=G[y]; G[y]=cnt;
}

int fa[N][21];

void dfs(int x,int f){
  l[x]=++t; dpt[x]=dpt[f]+1;
  fa[x][0]=f; for(int i=1;i<20;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  for(int i=G[x];i;i=E[i].nx)
    if(E[i].t!=f) dfs(E[i].t,x);
  r[x]=t;
}

inline int lca(int x,int y){
  if(dpt[x]<dpt[y]) swap(x,y);
  for(int i=19;~i;i--)
    if(dpt[fa[x][i]]>=dpt[y]) x=fa[x][i];
  if(x==y) return x;
  for(int i=19;~i;i--)
    if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}

int q,Q[N],k;

namespace IT{
  int G[N],cnt;
  struct edge{
    int t,nx,w;
  }E[N<<1];

  inline bool cmp(const int &a,const int &b){
    return l[a]<l[b];
  }

  int S[N],t;

  inline void addedge(int x,int y,int w){
    E[++cnt].t=y; E[cnt].nx=G[x]; E[cnt].w=w; G[x]=cnt;
    E[++cnt].t=x; E[cnt].nx=G[y]; E[cnt].w=w; G[y]=cnt;
  }

  int root,it[N],ts;

  inline int inTree(int x,int y){
    return l[x]<=l[y] && r[x]>=r[y];
  }

  void build(){
    sort(Q+1,Q+1+k,cmp);
    int nk=k; ++ts;
    for(int i=1;i<=k;i++) it[Q[i]]=ts;
    for(int i=1;i<k;i++)
      Q[++nk]=lca(Q[i],Q[i+1]);
    sort(Q+1,Q+1+nk,cmp); nk=unique(Q+1,Q+1+nk)-Q-1;
    t=cnt=0;
    for(int i=1;i<=nk;i++){
      int x=Q[i]; G[x]=0;
      while(t && !inTree(S[t],x)) t--;
      if(!t) root=x;
      else addedge(S[t],x,dpt[x]-dpt[S[t]]);;
      S[++t]=x;
    }
  }

  ll fst,sum[N];
  int size[N],g[N],h[N],snd,trd;

  void dfs(int x,int f){
    g[x]=0; sum[x]=0;
    if(it[x]==ts) size[x]=1,h[x]=0;
    else h[x]=1<<29,size[x]=0;
    int smax=0,smin=1<<29;
    for(int i=G[x];i;i=E[i].nx)
      if(E[i].t!=f){
    dfs(E[i].t,x);
    int cur=g[E[i].t]+E[i].w;
    if(cur>g[x]) smax=g[x],g[x]=cur;
    else smax=max(smax,cur);

    fst+=(sum[E[i].t]+1LL*E[i].w*size[E[i].t])*size[x]+sum[x]*size[E[i].t];
    sum[x]+=sum[E[i].t]+1LL*size[E[i].t]*E[i].w;
    size[x]+=size[E[i].t];

    cur=h[E[i].t]+E[i].w;
    if(cur<h[x]) smin=h[x],h[x]=cur;
    else smin=min(smin,cur);
      }
    snd=min(snd,h[x]+smin);
    trd=max(g[x]+smax,trd);
  }

  void solve(){
    build(); fst=trd=0; snd=1<<30;
    dfs(root,0);
    printf("%lld %d %d\n",fst,snd,trd);
  }
}

int main(){
  read(n);
  for(int i=1,x,y;i<n;i++)
    read(x),read(y),addedge(x,y);
  dfs(1,0); read(q);
  while(q--){
    read(k);
    for(int i=1;i<=k;i++) read(Q[i]);
    IT::solve();
  }
}