1127 ZigZagging on a Tree (30分)

后序与中序遍历构建二叉树

最新推荐文章于 2020-07-08 00:03:22 发布

原创最新推荐文章于 2020-07-08 00:03:22 发布 · 241 阅读

0 ·

CC 4.0 BY-SA版权

PAT 同时被 2 个专栏收录

61 篇文章

订阅专栏

二叉树

11 篇文章

订阅专栏

本文介绍了一种使用后序遍历和中序遍历确定二叉树层次遍历的方法。通过递归算法，首先从后序遍历中找到根节点，在中序遍历中定位该根节点，从而划分出左子树和右子树的范围。然后分别对左子树和右子树进行递归处理，最终实现层次遍历并按奇偶层数逆序或正序输出节点。

1.后序+中序确定层次遍历

2.用的是中序区间和后序区间来进行遍历（也可以用Tree Traversals (25分)中后序确定根结点+中序区间来做，因为后序区间也就可以判断根结点）

3.层次遍历看样例是第一行逆序，第二行正序，，以此类推所以用vector来存储每行的节点在判断是奇数层还是偶数层

#include<iostream>
#include<vector>

using namespace std;

const int N = 1000000+5;
int in[N], po[N];//中  后
vector<int> v[N];
int ans = 0;
void level(int il, int ir, int pol, int por, int l) {//先序遍历
//	cout << il << " " << ir << endl;
	if(il > ir) return;
	int i;

	for(i=il; i<=ir; i++) {
		if(in[i] == po[por]) {
			break;
		}
	}
//	cout << "得到的pos"<< i << endl;
	int num = i-il;//子树节点数
//	cout << l << "所对应的："<< po[por] << endl;
	level(il, i-1, pol, pol+num-1, l+1);//左子树
	v[l].push_back(po[por]);
	level(i+1, ir, pol+num, por-1, l+1);//右子树
	ans = max(ans, l);
}

int main() {
	int n;

	cin >> n;

	for(int i=0; i<n; i++) {
		cin >> in[i];//左右根
	}
	for(int i=0; i<n; i++) {
		cin >> po[i];//左根右
	}
	level(0, n-1, 0, n-1, 0);

	for(int i=0; i<=ans; i++) {
		if(i%2) {
			for(int j=0; j<v[i].size(); j++) {
				n--;
				printf("%d%c", v[i][j], " \n"[n==0]);
			}
		} else 	for(int j=v[i].size()-1; j>=0; j--) {
				n--;
				printf("%d%c", v[i][j], " \n"[n==0]);
			}

	}

	return 0;
}