算法之路之征服上海交大的oj-1040. 二叉树层次遍历

最新推荐文章于 2022-05-09 11:11:27 发布

原创最新推荐文章于 2022-05-09 11:11:27 发布 · 547 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

上海交大oj 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种通过广度优先搜索实现的二叉树层次遍历算法，并提供了完整的C++代码实现。该算法首先将根节点加入队列，然后依次处理队列中的每个节点，将其子节点加入队列并标记已访问状态。

二叉树层次遍历
Description

给出一棵二叉树，求它的层次遍历结果。

[二叉树的遍历问题是一种精神，务必领会]
Input Format

第一行，N<1000000，表示二叉树节点数。

默认序号为0的节点为树根。接下来共N-1行，
依次表示序号为1，…，N-1的节点的父亲节点序号。

如果一个节点有两个孩子节点，左孩子节点序号总是小于右孩子节点序号。
Output Format

仅一行，二叉树的层次遍历结果。节点序号间用空格隔开。
Hint
Sample Input

6
0
1
1
0
4

Sample Output

0 1 4 2 3 5

#include<iostream>
#include<vector>
#include<queue>
#include<string.h>
using namespace std;
const int MAXN = 1000005;
vector<int> G[MAXN];
int n;
queue<int> q;
bool vis[MAXN]; 
void addEdge(int a,int b){
    G[b].push_back(a); //建立树的过程
} 
int main(){
    cin>>n;
    memset(vis,false,sizeof(vis)); 
    for(int i=1;i<n;i++){
        int t;
        cin>>t;
        addEdge(i,t); 
    } 
    q.push(0);
    vis[0] = true;
    cout<<0<<" "; 
    while(!q.empty()){
        int u = q.front();
        q.pop();
        for(int i=0;i<G[u].size();i++){
            int v = G[u][i];
            if(!vis[v]){
                q.push(v);
                vis[v] = true; 
                cout<<v<<" ";
            } 
        } 
    }
    return 0;
}