2019江西省赛A题 HDU 6567 Cotree 树的重心 树上任意两点距离和

大意： n个点 n-2条边 生成了两颗树 现在要连一条边使其成为一棵树，并且树上任意两点间的距离和最小，输出这个距离和

思路： 跑两遍树的重心，将两棵树的重心连边，然后跑距离和

注意点：
两棵树的节点数(重心板子中用到的n）
并差集判断两点是否在一棵树上时用到的是find，不是father
双向边所以 maxn要开到2N

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=200005;
int n,m;
const int inf=0x3f3f3f3f;
int d[maxn];///大小数组
int head[maxn],cnt;///head->是总边数中的第几边，cnt->总边数
int nn1,nn2;///两棵树的节点数  注意重心模板中的n
int minnode,md;///两棵树的重心
int ans=inf;///每次跑重心之前  注意初始化

ll dis[maxn];///距离和
int father[maxn];///并查集方法
//int vis[maxn];///bfs方法标记是否和点1在一棵树上
void init()
{
    for(int i=0; i<maxn; i++)
        d[i]=0,head[i]=-1,dis[i]=0,father[i]=i;///,vis[i]=0;
    cnt=0;
    ans=inf;
}
//---------------------------初始化
struct edge
{
    int v,ne;
} ed[maxn];

void addedge(int u,int v)
{
    ed[cnt].v=v;
    ed[cnt].ne=head[u];///对于u点，上一条边的编号
    head[u]=cnt++;///这条边的编号,更新head
}
//---------------------------存边
int find(int x)
{
    if(father[x]==x)return x;
    else
    {
        father[x]= find (father[x]);   //加括号！！！
        return father[x];           //找祖先节点  压缩（不记录路径）
    }
}
void unite(int x,int y)
{
    int c=find(x);
    int d=find(y);
    if(c!=d)
        father[d]=c;   //y的祖先节点的父节点是x的祖先节点
}
//---------------------------并查集
/*
void bfs()
{
    queue<int>Q;
    vis[1]=1;
    Q.push(1);
    while(!Q.empty())
    {
        int x=Q.front();
        Q.pop();
        for(int i=head[x]; i!=-1; i=ed[i].ne)
        {
            int to=ed[i].v;
            if(vis[to])continue;
            vis[to]=1;
            Q.push(to);
        }
    }
}
*/ 
//---------------------------bfs
void dfs(int x,int fa,int nn)
{
    d[x]=1; //节点本身
    int maxSub=0; //maxSub为节点u的最大子树节点个数
    for(int i=head[x]; i!=-1; i=ed[i].ne)
    {
        int son=ed[i].v;
        if(son!=fa)
        {
            dfs(son,x,nn);
            d[x]+=d[son];
            maxSub=max(maxSub,d[son]);
        }
    }
    maxSub=max(maxSub,nn-d[x]);
    if(maxSub<ans)
    {
        minnode=x;
        ans=maxSub;
    }
}
//---------------------------重心
void dfs2(int x,int fa)
{
    d[x]=1;
    for(int i=head[x]; i!=-1; i=ed[i].ne)
    {
        int son=ed[i].v;
        if(fa==son)continue;
        dfs2(son,x);
        d[x]+=d[son];
        dis[x]+=dis[son]+1ll*d[son]*(n-d[son]);
    }
}
//---------------------------距离和
int main()
{

    while(cin>>n)
    {
        init();
        for(int i=1; i<n-1; i++)
        {
            int u,v;
            cin>>u>>v;
            unite(u,v);///并查集
            addedge(u,v);
            addedge(v,u);
        }
        
        int dian;///另一棵树中的一个点
        nn1=nn2=0;
        /*
        bfs();
        for(int i=1; i<=n; i++)nn1+=vis[i];
        nn2=n-nn1;
        */ ///bfs 点1所在的树有多少点
        
        int gd=find(1);
        for(int i=1;i<=n;i++)
        {
            if(find(i)==gd)nn1++;       ///注意这里是find  不是father！！！
            else dian=i;
        }///并查集 点1所在的树有多少点
        nn2=n-nn1;
        
        dfs(1,0,nn1);///nn1 两棵树节点数不同！！！！！
        md=minnode;///第一棵树的重心 md

        ans=inf;///注意初始化！！！！
        /*
        for(int i=1; i<=n; i++)
        {
            if(!vis[i])
            {
                dfs(i,0,nn2);
                break;
            }
        }
        */ ///bfs 另一棵树中的某点
        
        dfs(dian,0,nn2);///nn2 两棵树节点数不同！！！！！
        addedge(md, minnode);
        addedge(minnode, md);///两重心连边
        // cout<<md<<" "<<minnode<<endl;
        dfs2(1,-1);///统计距离和
        cout<<dis[1]<<endl;
    }
    return 0;
}