【PAT_1021】Deepest Root

题目描述:

连通非循环的图可以被认为是树。 树的高度取决于所选的根。 现在你应该找到导致最高树的根。 这样的根被称为最深的根。

输入:

每个输入文件包含一个测试用例。 对于每种情况,第一行包含正整数N(<= 10000),这是节点的数量,因此节点从1到N编号。然后N-1行跟随,每个都描述边缘给定两个 相邻节点的数字。

输出:

对于每个测试用例,在一行中打印每个最深的根。 如果这样的根不是唯一的,则按照其数字的递增顺序打印它们。 如果给定的图形不是树,则打印“Error:K components”,其中K是图中连接组件的数量。

解题思路:

实际上是n个节点之间的边能否构成一棵树,如果不能构成树输出连通分量的个数,能构成则得到构成最深树的高度并输出树的根节点,如果有多个就按从小到大的顺序输出。
所以本题可以先通过dfs判断连通分量的个数,如果只有一个,然后再近些n词dfs得到拥有最高高度的节点。

代码

#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
bool mark[10010];
int maxheigh = 0, resmax = 0;;
void dfs(vector<vector<int> >&graph, vector<int>&res,int start,int cur,int height) {
        mark[cur] = true;
        height++;
        if (graph[cur].size() == 1&& mark[graph[cur][0]]) {
            if (height > maxheigh) {
                maxheigh = height;
            }else if (height == maxheigh) {
                maxheigh = height;
            }
        }
        else {
            for (int i = 0; i < graph[cur].size(); i++) {
                if (!mark[graph[cur][i]]) {
                    dfs(graph, res,start, graph[cur][i],height);
                };
            }
        }
}
int main() {
        int n,a,b;
        cin >> n;
        vector<int>res;
        vector<vector<int> >graph(n + 1);
    fill(mark, mark + 10010, false);
        for (int i = 1; i < n; i++) {
            cin >> a >> b;
            graph[a].push_back(b);
            graph[b].push_back(a);
            
        }
        int components = 0;
        for (int i = 1; i <= n; i++) {
            if (!mark[i]) {
                dfs(graph, res,i,i,0);
                components++;
            }
        }
        if (components > 1) {
            printf("Error: %d components", components);
        }
        else {
            for (int i = 1; i <= n; i++) {
            fill(mark, mark + 10010, false);
                maxheigh = 0;
                dfs(graph, res, i, i, 0);
                if (maxheigh > resmax) {
                    res.clear();
                    res.push_back(i);
                    resmax = maxheigh;
                }   else if (resmax == maxheigh) {
                    resmax = maxheigh;
                    res.push_back(i);
                }
            }
            sort(res.begin(), res.end());
            for (int i = 0; i < res.size(); i++) {
                cout << res[i] << endl;
            }
        }
        return 0;
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 一些概念 数据结构就是研究数据的逻辑结构和物理结构以及它们之间相互关系,并对这种结构定义相应的运算,而且确保经过这...
    Winterfell_Z阅读 6,006评论 0 13
  • 因为之前就复习完数据结构了,所以为了保持记忆,整理了一份复习纲要,复习的时候可以看着纲要想具体内容。 树 树的基本...
    牛富贵儿阅读 7,025评论 3 10
  • 佟丽娅,新疆美女,中国女演员,毕业于中央戏剧学院表演系。 不要和消耗你的人在一起,也不要做消耗别人的人。永远不要在...
    一只草莓味的小可爱阅读 235评论 0 0
  • 大家好,我是Dany。今天给大家带来的这本书呢就是这本小白书《断舍离》。【现象】之所以想起来看这本书呢,是因为最近...
    DanyTan2333阅读 526评论 0 0