1020. Tree Traversals

最新推荐文章于 2022-07-04 14:31:24 发布

早迎朝阳晚迎星光

最新推荐文章于 2022-07-04 14:31:24 发布

阅读量640

点赞数

分类专栏： PAT 文章标签： pat acm 二叉树建立层次遍历

本文链接：https://blog.csdn.net/staibin/article/details/21297139

版权

PAT 专栏收录该内容

70 篇文章 0 订阅

订阅专栏

题目链接：http://pat.zju.edu.cn/contests/pat-a-practise/1020

二叉树建立、层次遍历

// 层次遍历
// 递归建立二叉树
// vector/queue

#include <stdio.h>
#include <iostream>
#include <queue>
#include <vector>

using namespace std;

struct Tree
{
	int key;
	Tree* left;
	Tree* right;
};

vector<int> level_order_buf;
queue<Tree*> q;

Tree *CreatTree(int in[], int post[], int n)
{
	if(n <= 0)
	{
		return NULL;
	}
	Tree *root = (Tree *)malloc(sizeof(Tree));
	root->key = post[n-1];
	root->left=NULL;
	root->right=NULL;
	int i;
	for(i=0;i<n; i++)
	{
		if(in[i] == post[n-1])
			break;
	}

	root->left=CreatTree(in, post, i);
	root->right=CreatTree(in+i+1, post+i, n-i-1);
	return root;
}

int levelOrder(Tree *root)
{
	
	while(q.empty() == false)
	{
		q.pop();
	}
	if(root != NULL)
	{
		q.push(root);
	}
	while(q.empty() == false)
	{
		Tree *node = q.front();
		q.pop();
		level_order_buf.push_back(node->key);
		if(node->left != NULL)
		{
			q.push(node->left);
		}
		if(node->right != NULL)
		{
			q.push(node->right);
		}
	}
	return 0;
}

int main()
{
	#ifdef ONLINE_JUDGE
	#else
		freopen("E:\\in.txt", "r", stdin);
	//	freopen("E:\\out.txt", "w", stdout);
	#endif

	int n;
	int i;
	int post[100], in[100];

	scanf("%d", &n);
	for(i=0;i<n;i++)
	{
		scanf("%d", &post[i]);
	}
	for(i=0;i<n;i++)
	{
		scanf("%d", &in[i]);
	}
	Tree *root = CreatTree(in, post, n);
	levelOrder(root);
	for(i=0;i<level_order_buf.size()-1;i++)
	{
		printf("%d ", level_order_buf[i]);
	}
	printf("%d\n", level_order_buf[i]);
	return 0;
}