[题解]树的遍历

wingaso

于 2022-02-23 16:05:12 发布

阅读量116

点赞数

分类专栏： C++ 题解文章标签： c++ 算法树结构数据结构

本文链接：https://blog.csdn.net/qq_35630119/article/details/123085090

版权

题解同时被 2 个专栏收录

55 篇文章 1 订阅

订阅专栏

C++

48 篇文章 4 订阅

订阅专栏

题目

题目描述

给出一个二叉树，请分别输出这棵二叉树的前序遍历、中序遍历和后序遍历结果。

样例输入

第一行输入就下点的个数 $n$ ，接下来 $n$ 行按照下述格式输入各结点的信息，每个结点占1行。

id left right

$i d$ 为结点编号， $l e f t$ 为左子结点编号， $r i g h t$ 为右子结点编号。不存在子结点时 $l e f t (r i g h t)$ 为-1。

样例输出

第一行按前序遍历的顺序输出结点编号。

第二行按中序遍历的顺序输出结点编号。

第三行按后序遍历的顺序输出结点编号。

题解

#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
using namespace std;
const int N = 1000;
int Son[N][2],Parent[N];
vector<int> Pre,In,Post;
void preOrder(int idx){
	Pre.push_back(idx);
	if(~Son[idx][0]) preOrder(Son[idx][0]);
	if(~Son[idx][1]) preOrder(Son[idx][1]);
}
void inOrder(int idx){
	if(~Son[idx][0]) inOrder(Son[idx][0]);
	In.push_back(idx);
	if(~Son[idx][1]) inOrder(Son[idx][1]);
}
void postOrder(int idx){
	if(~Son[idx][0]) postOrder(Son[idx][0]);
	if(~Son[idx][1]) postOrder(Son[idx][1]);
	Post.push_back(idx); 
}
int main(){
	int n;
	scanf("%d",&n);
	memset(Parent,-1,sizeof Parent);
	for(int i = 0;i < n;i++){
		int id,l,r;
		scanf("%d%d%d",&id,&l,&r);
		Son[id][0] = l;
		Son[id][1] = r;
		Parent[l] = Parent[r] = id;
	}
	int root = -1;
	for(int i = 0;i < n;i++) if(Parent[i] == -1) root = i;
	preOrder(root),inOrder(root),postOrder(root);
	
	for(int i = 0;i < n - 1;i++) printf("%d ",Pre[i]);
	printf("%d\n",Pre[n - 1]);
	for(int i = 0;i < n - 1;i++) printf("%d ",In[i]);
	printf("%d\n",In[n - 1]);
	for(int i = 0;i < n - 1;i++) printf("%d ",Post[i]);
	printf("%d\n",Post[n - 1]);
	return 0;
}

树的遍历理解

这道题考查了树遍历的三种常见的方式——前序遍历、中序遍历、后序遍历。

这里的前、中、后时对于根结点而言的，如上方三个函数所示。

这里三个函数都是递归函数，需要多加理解。

有没有非递归的写法呢？
有。非递归的写法缺点是相对于递归写法更加繁琐，而有点是在某些情况下可以避免递归爆栈情况的发生（如python语言的递归极容易爆栈，宜采用非递归的实现方式，C/C++语言一般使用递归实现即可）。

前序遍历的非递归实现如下所示：

void preOrder(int idx){
    stack<int> S;
    S.push(idx);
    while(!S.empty()){
        int t = S.top();
        S.pop();
        Pre.push_back(t);
        if(~Son[t][1]) S.push(Son[t][1]);
        if(~Son[t][0]) S.push(Son[t][0]);
    }
}