1021. Deepest Root (25)——DFS+并查集

http://pat.zju.edu.cn/contests/pat-a-practise/1021

无环连通图也可以视为一棵树，选定图中任意一点作为根，如果这时候整个树的深度最大，则称其为 deepest root。 给定一个图，按升序输出所有 deepest root。如果给定的图有多个连通分量，则输出连通分量的数量。

1.使用并查集判断图是否为连通的。

2.任意选取一点，做 dfs 搜索，选取其中一个最远距离的点 A，再做一次 dfs，找到的所有距离最远的点以及点 A 都是 deepest root。

考虑到为稀疏图，则使用动态链表

#include<stdio.h>
#include<stack>
#include<iostream>
#include<vector>
#include<set>
#include<queue>
using namespace std;

vector<int> map[];
int f[];
int find(int k){
    if(f[k]==-)return k;
    else
        return f[k]=find(f[k]);
}

int um(int a,int b){
    int fa,fb;
    fa=find(a);
    fb=find(b);

    if(fa==fb)return ;//表示有环

    if(fa!=fb)
        f[fa]=fb;
    return ;
}

int step[];

void dfs(int x,int STEP){

    int i;
    for(i=;i<map[x].size();i++){
        if(step[map[x][i]]!=)continue;
        step[map[x][i]]=STEP;
        dfs(map[x][i],STEP+);
    }
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        int i,a,b;

        for(i=;i<=n;i++){
            f[i]=-;
            step[i]=;
        }

        int huan=;
        for(i=;i<n;i++){
            scanf("%d%d",&a,&b);
            if(um(a,b)==)huan=;

            map[a].push_back(b);
            map[b].push_back(a);
        }
        int connectAdd=;
        set<int>set1;
        int ttemp;
        for(i=;i<=n;i++){
            ttemp=find(i);
            set1.insert(ttemp);
        }

        if(set1.size()>=||huan==){
            printf("Error: %d components\n",set1.size());continue;
        }
        step[]=;
        dfs(,);

        int max=,ri;
        for(i=;i<=n;i++){
            if(max<step[i]){
                max=step[i];
                ri=i;
            }
        }

        for(i=;i<=n;i++){
            step[i]=;
        }
        step[ri]=;
        dfs(ri,);

        max=;
        for(i=;i<=n;i++){
            if(max<step[i]){
                max=step[i];
            }
        }

        for(i=;i<=n;i++){
            if(step[i]==max||step[i]==){
                printf("%d\n",i);
            }
        }
    }

    return ;
}

其实上面的算法还有点问题，虽然AC了

考虑

5
1 2
1 3
2 4
2 5

应该是输出

3

4

5

算法改进 以（第2次dfs最深的点）为起点,再做DFS得到的最深的点，这些点才是所有最深的点

#include<stdio.h>
#include<stack>
#include<iostream>
#include<vector>
#include<set>
#include<queue>
using namespace std;

vector<int> map[];
int f[];
int find(int k){
    if(f[k]==-)return k;
    else
        return f[k]=find(f[k]);
}

int um(int a,int b){
    int fa,fb;
    fa=find(a);
    fb=find(b);

    if(fa==fb)return ;//表示有环

    if(fa!=fb)
        f[fa]=fb;
    return ;
}

int step[];
int deepF[];//第2次dfs最深的点
int deepR[];//以（第2次dfs最深的点）为起点,做DFS得到的最深的点

void dfs(int x,int STEP){

    int i;
    for(i=;i<map[x].size();i++){
        if(step[map[x][i]]!=)continue;
        step[map[x][i]]=STEP;
        dfs(map[x][i],STEP+);
    }
}

int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        int i,a,b,j;

        for(i=;i<=n;i++){
            f[i]=-;
            step[i]=;
            deepF[i]=;
            deepR[i]=;
        }

        int huan=;
        for(i=;i<n;i++){
            scanf("%d%d",&a,&b);
            if(um(a,b)==)huan=;

            map[a].push_back(b);
            map[b].push_back(a);
        }
        int connectAdd=;
        set<int>set1;
        int ttemp;
        for(i=;i<=n;i++){
            ttemp=find(i);
            set1.insert(ttemp);
        }

        if(set1.size()>=||huan==){
            printf("Error: %d components\n",set1.size());continue;
        }
        step[]=;
        dfs(,);

        int max=,ri;
        for(i=;i<=n;i++){
            if(max<step[i]){
                max=step[i];
                ri=i;
            }
        }

        for(i=;i<=n;i++){
            step[i]=;
        }
        step[ri]=;
        dfs(ri,);

        max=;
        for(i=;i<=n;i++){
            if(max<step[i]){
                max=step[i];
            }
        }

        for(i=;i<=n;i++){
            if(step[i]==max||step[i]==){
                //printf("%d\n",i);
                deepF[i]=;
                deepR[i]=;
            }
        }

        for(i=;i<=n;i++){
            if(deepF[i]==)continue;
            for(j=;j<=n;j++)step[j]=;

            dfs(i,);
            for(j=;j<=n;j++){
                if(step[j]==max)deepR[j]=;
            }
        }

        for(i=;i<=n;i++){
            if(deepR[i]==)printf("%d\n",i);
        }
    }

    return ;
}