PAT(Advanced)1021 Deepest Root (DFS)(并查集)(邻接表)C++实现

PAT(Advanced)甲级1021 Deepest Root (DFS)(并查集)(邻接表)C++实现

题目链接

PAT(Advanced)1021

题目大意

输入为无向无环图，任意顶点作为树根时能够达到树的最大深度的所有顶点（递增输出），同时，输入的图可能不连通，即无法构成一个树，此时输出该图树的数量（即根节点数目）。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int MAX_SIZE = 10000 + 10;
int root[MAX_SIZE];// 并查集，存放节点的祖先节点或根节点
bool visit[MAX_SIZE];// 用于DFS判断顶点是否已经被访问
int depth[MAX_SIZE];//存放各个顶点作为根节点的树可以达到的最大深度
int currentRoot;//用于标识当前树的根节点， 辅助更新depth数组的值
int maxDepth;//维护所有树的最大深度
vector<vector<int> > v;//邻接表
int FindSet(int x) {//并查集
	//若找到此节点的根节点（root[x] == x），返回根节点，否则从其父节点开始回溯，直到找到它的根节点，并更新路径上所有的节点的根节点
    return root[x] == x ? root[x] : root[x] = FindSet(root[x]);
}

void DFS(int node, int currentDepth) {
    visit[node] = true;// 标记该顶点已访问
    bool flag = true;// 立flag该节点为叶子节点
    for (int i = 0; i < v[node].size(); i++) {
        if (!visit[v[node][i]]) {// 该节点存在孩子节点，flag倒下
        	// DFS递归进入该节点的孩子节点
            flag = false;
            DFS(v[node][i], currentDepth + 1);
            visit[v[node][i]] = false;//为使得其他节点作为根节点可以顺利访问，在该节点遍历后，取消该节点的标记
        }
    }
    if (flag) {//flag不倒， 更新当前顶点最大深度，维护全局最大深度值
        depth[currentRoot] = max(depth[currentRoot], currentDepth);
        maxDepth = max(maxDepth, currentDepth);
    }
}

int main(int argc, char const *argv[])
{
    int N;
    scanf("%d", &N);
    vector<int> v0;
    v.push_back(v0);//使起始点为1
    bool tag[MAX_SIZE];//并查集标记数组
    for (int i = 1; i <= N; i++) {
        vector<int> t;
        v.push_back(t);
        root[i] = i;//并查集中，初始时每个节点的根节点为自身
        tag[i] = false;
        visit[i] = false;
        depth[i] = 0;
    }//初始化
    int node1, node2;
    for (int i = 1; i < N; i++) {
        scanf("%d%d", &node1, &node2);
        int node1Root = FindSet(node1);//查找节点的根节点
        int node2Root = FindSet(node2);
        root[node1Root] = node2Root;//更新节点的根节点，并为一棵树
        v[node1].push_back(node2);//构建无向图，邻接表
        v[node2].push_back(node1);
    }
    for (int i = 1; i <= N; i++) {//回溯求每个点的根节点
        FindSet(i);
        tag[root[i]] = true;//标记根节点
    }
    int K = 0;
    for (int i = 1; i <= N; i++) {
        if (tag[i]) {//累加存在的根节点数
            K++;
        }
    }
    if (K > 1) {//若存在多个根节点，错误处理
        printf("Error: %d components\n", K);
    } else {//若仅存在一个根节点，构成一棵树
        for (currentRoot = 1; currentRoot <= N; currentRoot++) {
            DFS(currentRoot, 0);//对每个节点为根节点进行深度优先遍历
            visit[currentRoot] = false;//遍历过后取消标记，以便下一次遍历
        }//currentRoot维护当前树的根节点
        for (int i = 1; i <= N; i++) {
            if (depth[i] == maxDepth) {//根据全局所有数最大深度，递增序输出节点
                printf("%d\n", i);
            }
        }
    }
    return 0;
}

算法思路

需先判断输入的图是否连通，利用并查集回溯求解每个节点的根节点，若所有节点的根节点相同则可以构成树，该图是连通的，若存在多个根节点，则该图存在多个连通分量，无法构成树，进入错误处理。判断图可以构成树之后，利用DFS算法，本题采用二维动态数组作为邻接表，从每一个点出发遍历全图，记录可以达到的最大深度，更新maxDepth，DFS遍历结束后，按照递增序输出所有符合条件的顶点。问题规模n下，本算法时间复杂度为O(n^2)，n个点每个点都进行一次时间复杂度O(n)的深度优先遍历。

样例输入1

5
1 2
1 3
1 4
2 5

样例输出1

3
4
5

样例输入2

5
1 3
1 4
2 5
3 4

样例输出2

Error: 2 components

鸣谢

感谢PAT提供的题目及测评平台

最后

需要注意的是，本题与DFS略有不同的是，需要进行多次遍历，深度优先遍历下，访问图中的每个节点一次且仅访问一次，本题需要访问多次，故需要在每次退出已访问节点及该节点为根节点的子树后，取消标记，以便下一个顶点为起点进行深度优先遍历。
由于博主水平有限，不免有疏漏之处，欢迎读者随时批评指正，以免造成不必要的误解！