L2-011 玩转二叉树(建树+BFS)

Posted MangataTS

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了L2-011 玩转二叉树(建树+BFS)相关的知识,希望对你有一定的参考价值。

题目链接

https://pintia.cn/problem-sets/994805046380707840/problems/994805065406070784

思路

由于给出了中根遍历以及先根遍历,我们先通过先根遍历定位到树的根节点,然后再通过中根遍历划分二叉树,然后再不断地往左往右递归处理左右子树,最后建立完这颗二叉树后,我们再在BFS 的时候优先把右子树放入队列,这样就能达到镜像反转的层序遍历,详情请看代码

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>
#define INF 0x3f3f3f3f

const int N = 1e2+10;

int in[N],pre[N],n;

struct Node
	int l,r;
Tree[N];

int build(int l1,int r1,int l2,int r2)//l1-r1中序遍历,l2-r2先序遍历
	if(l1 > r1 || l2 > r2) return 0;
	int p = l1;
	while(in[p] != pre[l2]) p++;
	int len = p - l1;
	int rt = pre[l2];
	Tree[rt].l = build(l1,p-1,l2+1,l2 + len);
	Tree[rt].r = build(p+1,r1,l2+len+1,r2);
	return rt;


void bfs(int s)
	queue<int> que;
	que.push(s);
	int cnt = 0;
	while(!que.empty())
		int rt = que.front();
		que.pop();
		cout<<rt<<" \\n"[++cnt==n];
		int l = Tree[rt].l,r = Tree[rt].r;
		if(r) que.push(r);
		if(l) que.push(l);
	



int main()

	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	cin>>n;
	for(int i = 1;i <= n; ++i) cin>>in[i];
	for(int i = 1;i <= n; ++i) cin>>pre[i];
	
	build(1,n,1,n);
	bfs(pre[1]);
	return 0;

以上是关于L2-011 玩转二叉树(建树+BFS)的主要内容,如果未能解决你的问题,请参考以下文章

L2-011. 玩转二叉树

L2-011 玩转二叉树 (25分)

L2-011玩转二叉树(25分)

L2-011. 玩转二叉树(PAT)

L2-011. 玩转二叉树

团体程序设计天梯赛 L2-006. 树的遍历 L2-011. 玩转二叉树