题目大意:
一颗二叉树有 N N N 个结点,编号从 0 ∼ N − 1 0 \sim N - 1 0∼N−1,先给出每个结点的左右结点编号(如果左右结点不存在用 − - − 表示)。现在请你输出该二叉树的反转二叉树的层序遍历和中序遍历。
思路:
这是一道非常典型的数据结构题。因为根节点没有父亲节点, 我们先利用输入找出根结点。
中序遍历的顺序是:左跟右, 那么反转二叉树对应原树中的顺序就是:右根左,然后利用
d
f
s
dfs
dfs 就能找出这个序列。
同理,一般我们求层序遍历都是用
b
f
s
bfs
bfs, 因为广搜的性质就是按照半径来搜,符合层序遍历的特点。
同理,广搜时反转二叉树左右结点对应原二叉树右左结点。因此先将右结点入队,然后左结点入队。
代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 15;
int n;
struct node{
int l, r;
}a[N];
bool st[N];
vector<int> ind, tra;
void inorder(int root){
if(root == -1) return;
if(a[root].r != -1) inorder(a[root].r);
ind.push_back(root);
if(a[root].l != -1) inorder(a[root].l);
}
int main(){
cin >> n;
for(int i = 0; i <n; i++){
char l, r;
cin >> l >> r;
if(isdigit(l)) a[i].l = l - '0', st[l - '0'] = true;
else a[i].l = -1;
if(isdigit(r)) a[i].r = r - '0', st[r - '0'] = true;
else a[i].r = -1;
}
int root;
for(int i = 0; i < n; i++){
if(st[i] == false){
root = i;
break;
}
}
inorder(root);
queue <int> q;
q.push(root);
while(q.size()){
int t = q.front();
q.pop();
tra.push_back(t);
if(a[t].r != -1) q.push(a[t].r);
if(a[t].l != -1) q.push(a[t].l);
}
for(int i = 0; i < n; i++){
if(i == 0) cout << tra[i];
else cout << " " << tra[i];
}
cout << endl;
for(int i = 0; i < n; i++){
if(i == 0) cout << ind[i];
else cout << " " << ind[i];
}
return 0;
}