#广度优先遍历之树的高度#南京理工大学机试

https://www.acwing.com/problem/content/3702/

思路:本题明面上是求树的高度,但却给出了类似于邻接表的存储方式,类似于一张无环图,但完全不知道节点之间的父子关系。考虑两种搜索方式,层序遍历可以严格保证先父后子的遍历顺序,故采用层序遍历。同时设置一个辅助数组来记录节点是否被访问过,如有,则证明是当前节点的父辈。

做法:分别定义一个vector<vector<int>>型的邻接表结构,一个存储遍历顺序的队列toVisit,一个记录是否访问和最大长度的数组distance,循环访问邻接表。初始将固定的根节点压入队列。当队列不为空时:

1.读取队首元素并出队

2.遍历队首元素所对应的vector,对于每一个节点:

a.若对应distance值不为-1,跳过

b.若为-1,证明尚未被访问。将该节点压入队列末尾,同时将distance设为从根到当前节点的父节点的值+1,并更新max

3.直到循环退出,打印max即可

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n,m;
    int max=0;
    scanf("%d%d",&n,&m);
    vector<vector<int>> table(n+1);
    for(int i=0;i<n-1;++i){
        int u,v;
        scanf("%d%d",&u,&v);
        table[u].push_back(v);
        table[v].push_back(u);
    }
    queue<int> toVisit;
    vector<int> distance(n+1);
    for(int i=1;i<=n;++i) distance[i] = -1;
    distance[m]=0;
    toVisit.push(m);

    while(!toVisit.empty()){
        int current = toVisit.front();
        toVisit.pop();

        for(int i=0;i<table[current].size();i++){
            int child = table[current][i];
            if(distance[child] != -1){
                continue;
            }
            toVisit.push(child);
            distance[child] = distance[current] + 1;
            max=distance[child];
        }
    }
    printf("%d\n",max);
    return 0;
}

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-01 17:13
想去,但是听说加班强度实在难崩,所以拒绝了,现在有点心梗对面hr感觉也是实习生,打电话的时候怪紧张的,但是感觉人很好嘞
水中水之下水道的鼠鼠:哥们这不先去体验一下,不行再跑呗,大不了混个实习经历(有更好的转正offer就当我没说)
点赞 评论 收藏
分享
06-20 19:40
中原工学院 Java
网络存储:十几天不会让你拉人办卡就结束了吧?
点赞 评论 收藏
分享
程序员牛肉:主要是因为小厂的资金本来就很吃紧,所以更喜欢有实习经历的同学。来了就能上手。 而大厂因为钱多,实习生一天三四百的就不算事。所以愿意培养你,在面试的时候也就不在乎你有没有实习(除非是同级别大厂的实习。) 按照你的简历来看,同质化太严重了。项目也很烂大街。 要么换项目,要么考研。 你现在选择工作的话,前景不是很好了。
点赞 评论 收藏
分享
评论
1
2
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务