bzoj 3611: [Heoi2014]大工程 虚树+树形dp

题意

国家有一个大工程，要给一个非常大的交通网络里建一些新的通道。
我们这个国家位置非常特殊，可以看成是一个单位边权的树，城市位于顶点上。
在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径。
现在国家有很多个计划，每个计划都是这样，我们选中了 k 个点，然后在它们两两之间 新建 C(k,2)条 新通道。
现在对于每个计划，我们想知道：
1.这些新通道的代价和
2.这些新通道中代价最小的是多少
3.这些新通道中代价最大的是多少
n<=1000000 ，q<=50000并且保证所有k之和<=2*n

分析

看到诸如保证所有k之和<=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;
}