hdu6567 Cotree（树形dp）

Cotree

本题大意：给出两棵树，让两个树随意连在一起，使连在一起的树的每个点到其他点的总和最小。

可以通过分析得到，想让每个点到其他点的总和最小，那必须是连接的重心，所以本题分为两个过程，找重心连在一起，在求出，每个点到其他点的总和，每个点到其他点的距离之和其实可以通过加每条边的使用次数来得到，所以就完事了。

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>
#include <stack>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int s1,s2,po1,po2,vis[maxn];
ll n,sum,dep[maxn],ma[maxn];
vector<int>a[maxn];
void dfs(int x,int fa)//求总和
{
    dep[x]=1;
    for(int i=0; i<a[x].size(); i++)
    {
        int y=a[x][i];
        if(y==fa)continue;
        dfs(y,x);
        dep[x]+=dep[y];
        sum+=(dep[y]*(n-dep[y]));//这条边的左部分的点*右部分的点就是使用次数
    }
}
void dp1(int x,int fa)//求第一棵树的重心
{
    s1++,vis[x]=1,dep[x]=1;
    for(int i=0; i<a[x].size(); i++)
    {
        int y=a[x][i];
        if(y==fa)continue;
        dp1(y,x);
        dep[x]+=dep[y];
        ma[x]=max(ma[x],dep[y]);
    }
}
void dp2(int x,int fa)//求第二棵树的重心
{
    dep[x]=1;
    for(int i=0; i<a[x].size(); i++)
    {
        int y=a[x][i];
        if(y==fa)continue;
        dp2(y,x);
        dep[x]+=dep[y];
        ma[x]=max(ma[x],dep[y]);
    }
}
int main()
{
    ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    cin>>n;
    for(int i=1; i<n-1; i++)
    {
        int u,v;
        cin>>u>>v;
        a[u].push_back(v),a[v].push_back(u);
    }
    memset(dep,0,sizeof(dep)),memset(ma,0,sizeof(ma));
    s1=0;
    dp1(1,-1);
    s2=n-s1;
    ll mm=inf;
    for(int i=1; i<=n; i++)
        if(vis[i])
        {
            ma[i]=max(ma[i],s1-dep[i]);
            if(ma[i]<mm)mm=ma[i],po1=i;
        }
    for(int i=1; i<=n; i++)
        if(!vis[i])
        {
            memset(dep,0,sizeof(dep)),memset(ma,0,sizeof(ma));
            dp2(i,-1);
            break;
        }
    mm=inf;
    for(int i=1; i<=n; i++)
       if(!vis[i])
        {
            ma[i]=max(ma[i],s2-dep[i]);
            if(ma[i]<mm)mm=ma[i],po2=i;
        }
    a[po1].push_back(po2),a[po2].push_back(po1);
    sum=0;
    memset(dep,0,sizeof(dep));
    dfs(1,-1);
    cout<<sum<<endl;
    return 0;
}