UVALive - 8072 - Keeping On Track（树形dp）

题目链接：https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=6094

题意：给你n(n<=10000)条无向边，这n条边刚好构成n+1个节点的树，你要删除某个节点及其相连的边，使变为不能互相到达的点对数最多，输出最多的点对数，然后在删完的基础上再添加一条边，让尽可能多的点对能相互到达，输出剩下的不能相互到达的点对数。

思路：

设dp[u]为以u为根的子树的大小，则

sum=0;

ans1+=dp[v]*(n-dp[u]+sum);

sum+=dp[v];

对所有的u=1,2,...,n,v为u的所有儿子节点，取一个最大值。想想为什么。

ans2即为ans1-删除上面的点之后剩下的连通块中点的数目最多的两个连通块的乘积

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(register int i=(a);i<=(b);i++)
#define dep(i,a,b) for(register int i=(a);i>=(b);i--)
using namespace std;
const int maxn=20010;
const int maxm=80010;
int n,m,k;
int cnt,id1;
ll ans,tmp;
int he[maxn],ok[maxn];
int sz[maxn];
int aa[maxn],ct;
struct node
{
    int v,nxt;
}a[maxm];
void add(int u,int v)
{
    a[cnt].v=v;
    a[cnt].nxt=he[u];
    he[u]=cnt++;
}
void init()
{
    cnt=0;
    memset(he,-1,sizeof(he));
    memset(sz,0,sizeof(sz));
    memset(ok,0,sizeof(ok));
    tmp=0;ans=0;
}
void dfs(int u,int fa)
{
    ll tp=0,sm=0;
    sz[u]=1;
    for(int i=he[u];i!=-1;i=a[i].nxt)
    {
        int v=a[i].v;
        if(v==fa) continue;
        dfs(v,u);
        sz[u]+=sz[v];
    }
    for(int i=he[u];i!=-1;i=a[i].nxt)
    {
        int v=a[i].v;
        if(v==fa) continue;
        tp+=sz[v]*(n-sz[u]+sm);
        sm+=sz[v];
    }
    if(tp>tmp)
    {
        id1=u;
        tmp=tp;
    }
}
void dfs2(int u,int fa,int rt)
{
    sz[rt]++;ok[u]=1;
    for(int i=he[u];i!=-1;i=a[i].nxt)
    {
        int v=a[i].v;
        if(ok[v]) continue;
        dfs2(v,u,rt);
    }
}
int main()
{
    int T,cas;
    while(scanf("%d",&n)!=EOF)
    {
        n++;
        init();
        rep(i,1,n-1)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            x++;y++;
            add(x,y);
            add(y,x);
        }
        dfs(1,-1);
        ok[id1]=1;
        memset(sz,0,sizeof(sz));
        ct=0;
        aa[ct++]=0;aa[ct++]=0;
        rep(i,1,n)
        if(!ok[i]){
            dfs2(i,-1,i);
            aa[ct++]=sz[i];
        }
        sort(aa,aa+ct);
        ans=(ll)aa[ct-1]*(ll)aa[ct-2];
        printf("%lld %lld\n",tmp,tmp-ans);
    }
    return 0;
}