笔记——二叉树遍历

题目

输入一棵二叉树，输出树的前、中、后序遍历结果。

输入一个整数N（N<= 10000)，表示树中有N个结点（编号0~N-1）。

接下来N行，依次为结点0~结点N-1的左右孩子情况。

每行3个整数，F,L,R。L,R为F的左右孩子。L,R如果为-1表示该位置上没有孩子。

分三行分别输出树的前中后序遍历。

同一行中的数字，用一个空格间隔。

输入样例

5

0 3 1

1 2 -1

2 -1 4

3 -1 -1

4 -1 -1

输出样例

0 3 1 2 4

3 0 2 4 1

3 4 2 1 0

 

代码如下：

#include<iostream>
#include<queue>
using namespace std;
#define maxSize 100
struct BTNode
{
	BTNode* parent;
	BTNode* lchild;
	BTNode* rchild;
	int data;
};

BTNode* createBT(int n,int* que)
{
	//1.把二叉树各个结点建立起来
	//2.把data赋值给各个二叉树结点
	BTNode* QNode[maxSize];
	int j=0;
	//queue<BTNode*> QNode;
	for(int i=0;i<3*(n-1)+1;i=i+3)
	{
		BTNode* node=new BTNode();
		BTNode* left=new BTNode();
		BTNode* right=new BTNode();
		node->data=que[i];
		left->data=que[i+1];
		right->data=que[i+2];
		node->lchild=left;
		node->rchild=right;
		QNode[j++]=node;
	}

	//在各个结点中寻找关系
	for(int m=0;m<n;m++)
	{
		for(int k=0;k<n;k++)
		{
			if(QNode[m]->lchild->data==QNode[k]->data)
			{
				QNode[m]->lchild=QNode[k];
				QNode[k]->parent=QNode[m];
			}
			else if(QNode[m]->rchild->data==QNode[k]->data)
			{
				QNode[m]->rchild=QNode[k];
				QNode[k]->parent=QNode[m];
			}
		}
	}
	//寻找根节点
	int a=0;
	for(;a<n;a++)
	{
		if(!QNode[a]->parent)
			break;
	}
	return QNode[a];
}
void preOrder(BTNode* root)
{
	if(root)
	{
		if(root->data!=-1)
			cout<<root->data<<" ";
		preOrder(root->lchild);
		preOrder(root->rchild);
	}
}
void inOrder(BTNode* root)
{
	if(root)
	{
		inOrder(root->lchild);
		if(root->data!=-1)
			cout<<root->data<<" ";
		inOrder(root->rchild);
	}
}
void postOrder(BTNode* root)
{
	if(root)
	{
		postOrder(root->lchild);
		postOrder(root->rchild);
		if(root->data!=-1)
			cout<<root->data<<" ";
	}
}
int main()
{
	int que[maxSize];
	int n;
	
	cin>>n;
	BTNode* root;
	
	for(int i=0;i<3*n;i++)
	{
		cin>>que[i];
	}
	root=createBT(n,que);
	preOrder(root);
	cout<<endl;
	inOrder(root);
	cout<<endl;
	postOrder(root);
	cout<<endl;
	system("pause");
	return 0;
}