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)的主要内容,如果未能解决你的问题,请参考以下文章