【PAT Advanced Level】1021. Deepest Root (25)

这题分两个步骤：

1. 求出图中有几个连通子图

2. 如果只有一个连通图，找出最长的路径（图的直径）

我直接用了一个DFS求连通子图的个数，其实更通用的方法是并查集。

找出最长路径，我们进行了n次DFS，每次以一个节点作为根节点。后来看了网上的解法，其实只需要两次DFS就可以求得。

昨天我刚刚看了怎么求树的直径，今天该用到的时候又忘了，真是学的不牢啊。具体求法见http://blog.csdn.net/gzxcyy/article/details/14165829

//一次DSF即可#include <iostream>#include <vector>#include <algorithm>#include <fstream>using namespace std;int num;vector<vector<int>> adj;vector<int> visited;vector<int> deepest;vector<int> distances;int connected = 0;void DSF_VISIT(int i){visited[i] = 1;for(int j = 0; j < adj[i].size(); j++){if(!visited[adj[i][j]]){distances[adj[i][j]] = distances[i] + 1;DSF_VISIT(adj[i][j]);}}visited[i] = 2;}void DSF(){for(int i = 1; i <= num; i++){if(!visited[i]){DSF_VISIT(i);connected++;}}}int main(){//fstream cin("a.txt");cin>>num;int tmp = num - 1;adj.assign(num + 1, vector<int>());visited.assign(num + 1, 0);deepest.assign(num + 1, 0);distances.assign(num + 1, 0);int a,b;while (tmp--){cin>>a>>b;adj[a].push_back(b);adj[b].push_back(a);}DSF();if(connected > 1)cout<<"Error: "<<connected<<" components"<<endl;else{for(int i = 1; i <= num; i++){visited.assign(num + 1, 0);distances.assign(num + 1, 0);DSF_VISIT(i);vector<int>::iterator it = max_element(distances.begin(), distances.end());deepest[i] = *it;}vector<int>::iterator it = max_element(deepest.begin(), deepest.end());for(int i = 1; i < deepest.size(); i++){if(deepest[i] == *it)cout<<i<<endl;}}}