PAT 甲级 1020

最新推荐文章于 2022-07-14 15:42:09 发布

小实在工大

最新推荐文章于 2022-07-14 15:42:09 发布

阅读量288

点赞数

分类专栏： PAT题解文章标签： c++ 算法开发语言

本文链接：https://blog.youkuaiyun.com/m0_61974219/article/details/121440371

版权

PAT题解专栏收录该内容

22 篇文章

订阅专栏

这篇博客介绍了如何解决PAT甲级1020题目，该题要求根据二叉树的中序和后序遍历结果，输出层序遍历。博主提出利用后序遍历的最后一个元素作为根节点，在中序遍历中找到其位置，将中序遍历分为左右子树，并递归处理。在递归过程中，通过标记进行重新排序，最终按层序输出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

PAT1020

题意：给出一颗二叉树的中序和后序遍历，让你输出层序遍历，给出的值都是互不相等的正整数。

思路：因为后序的最后一个总是根结点，令i在中序中找到该根结点，则i把中序分为两部分，左边是左子树，右边是右子树，于是可以递归地去处理。在递归的过程中加一个标记用于重新排序，按照层序重新排序后输出即可

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

struct node 
{
	int index;
	int val;
};

bool cmp(node a, node b) 
{
	return a.index<b.index;
}

vector<int> post,in;
vector<node> ans;

void pre(int root, int start, int end, int index) 
{
	if(start>end) 
		return;
	int i=start;
	while(in[i]!=post[root]&&i<end) 
		i++;
	ans.push_back({index, post[root]});
	pre(root-1-end+i,start,i-1,2*index+1);
	pre(root-1,i+1,end,2*index+2);
}

int main() 
{
	int n;
	scanf("%d",&n);
	post.resize(n);
	in.resize(n);
	for(int i=0;i<n;i++) 
		scanf("%d",&post[i]);
	for(int i=0;i<n;i++) 
		scanf("%d",&in[i]);
	pre(n-1,0,n-1,0);
	sort(ans.begin(),ans.end(),cmp);
	for(int i=0;i<ans.size();i++) 
	{
		if(i!=0)
			cout<<" ";
		cout<<ans[i].val;
	}
}