自底向上遍历二叉树

自底向上遍历二叉树

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

给定一棵包含N个节点的二叉树，每个节点编号1~N。特别的，如果u和v(u < v)是x的两个儿子节点，我们将其中较小的节点u视为x的左儿子，较大的节点v视为x的右儿子。例如下图3是4的左儿子，7是4的右儿子。

4
/ \
3 7
现在希望你从最左的叶子节点开始依次自底向上遍历，对于每一个叶子向上遍历到根，并且在遍历过程中忽略已经遍历过的节点。

对于两个叶子节点X和Y，我们是这样判断哪个更靠左：找到X和Y的最近公共祖先A，此时X和Y必然一个在A的左子树，另一个在A的右子树。我们认为在左子树的叶子节点更靠左。

例如对于下图

1
/ \
2 3
/ \
4 5
/ \ \
6 7 8
遍历顺序是:2-1-6-4-3-7-8-5

输入

第一行包含一个整数N。(1 ≤ N ≤ 100000)

以下N-1行每行包含两个整数u和v，代表u是v的父节点。

输出

按照遍历顺序输出N个节点的编号，每行一个。

样例输入

8
1 2
1 3
3 4
3 5
4 6
4 7
5 8

样例输出

2
1
6
4
3
7
8
5

代码片

#include <bits/stdc++.h>
using namespace std;
const int MAXN=1e5+100;
struct node
{
    int par=0;//父节点,初始化0默认无父节点
    vector<int>son;//子节点
}nod[MAXN];
queue<int>q;//统计0度叶子节点
bool vis[MAXN];
void dfs(int r)
{//深度优先遍历子节点统计0度叶子节点
    if(nod[r].son.size()==0){q.push(r);return;}
    if(nod[r].son.size()==2 && nod[r].son[0]>nod[r].son[1]) 
        swap(nod[r].son[0],nod[r].son[1]);//按题目要求，左右节点交换
    for(int i=0;i<nod[r].son.size();i++) dfs(nod[r].son[i]);
}
int main()
{
    int N;
    scanf("%d",&N);
    for(int i=0;i<N-1;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        nod[v].par=u;
        nod[u].son.push_back(v);
    }
    for(int i=1;i<=N;i++){//找根节点
        if(nod[i].par==0){dfs(i);break;}
    }
    while(!q.empty())
    {
        int p=q.front();q.pop();
        while(!vis[p] && p){
            printf("%d\n",p);
            vis[p]=true;
            p=nod[p].par;
        }
    }
    return 0;
}