PAT L2-011 玩转二叉树
转载链接
如果可以的话,可以点链接打赏一下这位学姐,写的挺不错的。
给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(<=30),是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7
输出样例:
4 6 1 7 5 3 2
跟树的遍历这道题目类似,所谓的镜面反转就是先遍历右子树后再遍历左子树。代码相似,注意理解,特别是两个递归分支的参数怎么写,在纸上模拟一遍就行了。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdlib>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <bitset>
using namespace std;
vector<int>pre,mid,level(10000,-1);
void post(int root,int start,int end,int index){
if(start>end) return ;
int i = start;
while(i<end&&mid[i]!=pre[root])i++;
level[index] = pre[root];
post(root+1,start,i-1,2*index+2);
post(root+1+i-start, i+1,end , 2*index+1);
}
int main(int argc, const char * argv[]) {
int n;
cin>>n;
mid.resize(n);
pre.resize(n);
for(int i=0;i<n;i++){
cin>>mid[i];
}
for(int i=0;i<n;i++){
cin>>pre[i];
}
int cnt = 0;
post(0,0,n-1,0);
for(int i=0;i<level.size();i++){
if(level[i]!=-1&&i!=0){
printf(" %d",level[i]);
cnt++;
}
else if(level[i]!=-1){
printf("%d",level[i]);
}
}
return 0;
}