bzoj 3611 [Heoi2014]大工程 虚树DP

题意就不说了，比较复杂，自己看吧。

分析：
我还是naive了，居然想着直接做。。以后看见这种询问个数之和<=n的直接无脑上虚树。
至于虚树跟之前的那个消耗战差不多。但是dp就很复杂了，不好做，要把每个点单独
拿出来做。具体转移看标程吧。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 1000005
#define ll long long
#define inf 0x3fffffff
using namespace std;

int cnt,n,m,last[N],ls[N],dep[N],fa[N][20],size[N],mx[N],mn[N],ans2,ans3,tim,dfn[N],logn,a[N],flag[N],st[N];
ll sum[N],ans1;
struct edge{int to,next;}e[N*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge1(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;
}

void addedge2(int u,int v)
{
    if (u==v) return;
    e[++cnt].to=v;e[cnt].next=ls[u];ls[u]=cnt;
}

void dfs(int x)
{
    dfn[x]=++tim;
    for (int i=1;i<=logn;i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    dep[x]=dep[fa[x][0]]+1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x][0]) continue;
        fa[e[i].to][0]=x;
        dfs(e[i].to);
    }
}

int getlca(int x,int y)
{
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=logn;i>=0;i--)
        if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    for (int i=logn;i>=0;i--)
        if (fa[x][0]!=fa[y][0])
        {
            x=fa[x][0];y=fa[y][0];
        }
    return fa[x][0];
}

bool cmp(int a,int b)
{
    return dfn[a]<dfn[b];
}

void dp(int x)
{
    int mx1=0,mx2=0,mn1=inf,mn2=inf;size[x]=0;sum[x]=0;
    if (flag[x])
    {
        size[x]++;sum[x]=dep[x];mn1=0;
    }
    for (int i=ls[x];i;i=e[i].next)
    {
        dp(e[i].to);
        size[x]+=size[e[i].to];sum[x]+=sum[e[i].to];
        if (mx[e[i].to]+dep[e[i].to]-dep[x]>mx1)
        {
            mx2=mx1;mx1=mx[e[i].to]+dep[e[i].to]-dep[x];
        }
        else if (mx[e[i].to]+dep[e[i].to]-dep[x]>mx2) mx2=mx[e[i].to]+dep[e[i].to]-dep[x];
        if (mn[e[i].to]+dep[e[i].to]-dep[x]<mn1)
        {
            mn2=mn1;mn1=mn[e[i].to]+dep[e[i].to]-dep[x];
        }
        else if (mn[e[i].to]+dep[e[i].to]-dep[x]<mn2) mn2=mn[e[i].to]+dep[e[i].to]-dep[x];
    }
    for (int i=ls[x];i;i=e[i].next)
        ans1+=(ll)(size[x]-size[e[i].to])*(sum[e[i].to]-(ll)size[e[i].to]*dep[x]);
    ans2=min(ans2,mn1+mn2);if (flag[x]||!flag[x]&&mx2) ans3=max(ans3,mx1+mx2);
    mx[x]=mx1;mn[x]=mn1;
    ls[x]=0;
}

void solve()
{
    cnt=0;
    int k=read();
    for (int i=1;i<=k;i++)
    {
        a[i]=read();
        flag[a[i]]=1;
    }
    sort(a+1,a+k+1,cmp);
    int top=1;
    st[1]=1;
    for (int i=1;i<=k;i++)
    {
        int now=a[i],f=getlca(now,st[top]);
        while (1)
        {
            if (dep[f]>=dep[st[top-1]])
            {
                addedge2(f,st[top]);top--;
                if (st[top]!=f) st[++top]=f;
                break;
            }
            addedge2(st[top-1],st[top]);top--;
        }
        if (st[top]!=now) st[++top]=now;
    }
    while (top>1)
    {
        addedge2(st[top-1],st[top]);top--;
    }
    ans1=ans3=0;ans2=inf;
    dp(1);
    for (int i=1;i<=k;i++)
        flag[a[i]]=0;
    printf("%lld %d %d\n",ans1,ans2,ans3);
}

int main()
{
    n=read();
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read();
        addedge1(x,y);
    }
    logn=log(n)/log(2);
    dfs(1);
    m=read();
    for (int i=1;i<=m;i++)
        solve();
    return 0;
}