pat1021

pat1021

1.题目大意

求无向图的最大深度，如果不是树，那么就输出联通分量。

记录使用之前的add函数来加边，由于是无向图，需要两边都加

同时，使用dfs来进行统计树的高度，还有一件事，就是因为是无向图，所以，要避免自己再次出现。

int dfs(int u,int father){
//    由于是无向图，难免会把自己加入的，father是自己，就需要跳过
    int depth =0;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == father) continue;
        depth = max(depth, dfs(j, u) + 1);
    }
    return depth;

}

因为这是图，所以以前是左右子树的进行比较，现在是对所有的节点来进行比较。

同时还有一个是联通分量的求法，就是使用并查集

通过对不是同一个图的进行加并查集，可以查出有几个联通部分

if(find(a)!= find(b)){
            k--;
            p[find(a)]= find(b);
//            进行连接
        }

3.代码

#include <cstring>
#include <iostream>
#include <algorithm>
#include "unordered_map"
#include "queue"
using namespace std;

const int N = 10010,M=2*N;

//使用连通图来判断是否有环，，同时使用ha来加之

int h[N],ne[M],e[M],idx;
//默认idx是0
int p[N];
//这是并查集验证，p【i】=i是高跟节点
int find(int x){
//    while (p[x]!=x){
//        x=p[x];
//    }
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

//定义a点加入b的链表
void add(int a, int b){
    e[idx]=b;
    ne[idx]=h[a];
//    加入到a的链表里面
    h[a]=idx;
//    更新a的队列地址
    idx++;
//    给下个节点开始的位置
}
int n;
vector<int> v1;
int  maxdepth=0;
int dfs(int u,int father){
//    由于是无向图，难免会把自己加入的，father是自己，就需要跳过
    int depth =0;
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == father) continue;
        depth = max(depth, dfs(j, u) + 1);
    }
    return depth;

}
int main()
{
    cin>>n;
//    初始化所有 s子节点的h【a=1
    memset(h,-1,sizeof h);
//    初始化并查集
    for (int i = 1; i <=n; ++i) {
        p[i]=i;
    }
    int k=n;
//    记录联通的
    for(int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
//        首先查询是否联通
        if(find(a)!= find(b)){
            k--;
            p[find(a)]= find(b);
//            进行连接
        }
        add(a,b);
        add(b,a);
    }
    if (k > 1) printf("Error: %d components", k);
    else{
//        进行dfs输出
        dfs(1,-1);
        for (auto v : v1) cout << v << endl;
    }
//    便利所有的节点，找出最大的
    for (int i = 1; i <=n; ++i) {
        int depth = dfs(i, -1);
//        都是父节点开始
        if (depth > maxdepth)
        {
            maxdepth = depth;
            v1.clear();
            v1.push_back(i);
        }
        else if (depth == maxdepth)
            v1.push_back(i);
    }

    for (auto v : v1) cout << v << endl;

    return 0;
}