目录
1,题目描述
Sample Input 1:
7
1 2 3 4 6 7 5
2 6 7 4 5 3 1
Sample Output 1:
Yes
2 1 6 4 7 3 5
Sample Input 2:
4
1 2 3 4
2 4 3 1
Sample Output 2:
No
2 1 3 4
题目大意
根据二叉树的先序遍历和后序遍历,输出一种中序遍历。中序遍历唯一输出Yes,否则输出No。
2,思路
参考大神@日沉云起【pat甲级1119. Pre- and Post-order Traversals (30)】
关键点
1,由先序遍历和后序遍历推出树时,当一棵树只有左子树/右子树时,此树的形态不唯一;
先序:1,2,3,4,5
后序:5,4,3,2,1
2,充分利用先序遍历和后序遍历的特点:
先序遍历的第一个节点和后序遍历的最后一个节点是根节点;
紧跟在先序遍历根节点后的第一个节点为左子树的根节点;
找到左子树的根节点在后序遍历中的位置,就可以找到左/右子树的起始点在后序遍历中的位置,进而实现递归;
3,先根序列中根节点的下一结点在后根序列中的位置正好等于right-1,即树只有左子树/右子树;
4,什么时候将节点放入中序遍历序列?
当只有一个叶节点时(left==right);
当左子树遍历完毕时;
3,AC代码
#include<bits/stdc++.h>
using namespace std;
int pre[30], post[30];
vector<int> in;
bool flag = true;
void inOrder(int root, int left, int right){ //root针对先序遍历 left/right针对后序遍历
if(left >= right){
if(left == right){
in.push_back(pre[root]); // !!!pre[root]不是root
}
return;
}
int i = left; // !!!不是int i = 0;
while(i <= right-1 && post[i] != pre[root+1]) // !!!不是while(i <= right && post[i] != pre[root+1])
i++;
if(i == right - 1) //只有左子树/右子树
flag = false;
inOrder(root+1, left, i); // !!!不是inOrder(root+1, left, i-1);
in.push_back(pre[root]); // !!!pre[root]不是root
inOrder(root+i-left+1+1, i+1, right-1); // !!!不是inOrder(root+i-left+1+1, i, right-1);
}
int main(){
#ifdef ONLINE_JUDGE
#else
freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE
int N;
cin>>N;
for(int i = 0; i < N; i++)
scanf("%d", &pre[i]);
for(int i = 0; i < N; i++)
scanf("%d", &post[i]);
inOrder(0, 0, N-1);
printf("%s\n",flag==true?"Yes":"No");
for(int i=0;i<in.size();++i)
printf("%s%d",i>0?" ":"",in[i]);
printf("\n");
return 0;
}
4,解题过程
第一搏
跟着大佬的代码敲了一遍。。。
#include<bits/stdc++.h>
using namespace std;
int pre[30], post[30];
vector<int> in;
bool flag = true;
void inOrder(int root, int left, int right){//root针对先序遍历 left/right针对后序遍历
if(left >= right){
if(left == right){
in.push_back(pre[root]);// !!!pre[root]不是root
}
return;
}
int i = left;// !!!不是int i = 0;
while(i <= right-1 && post[i] != pre[root+1])// !!!不是while(i <= right && post[i] != pre[root+1])
i++;
if(i == right - 1)//只有左子树/右子树
flag = false;
inOrder(root+1, left, i);// !!!不是inOrder(root+1, left, i-1);
in.push_back(pre[root]);// !!!pre[root]不是root
inOrder(root+i-left+1+1, i+1, right-1);// !!!不是inOrder(root+i-left+1+1, i, right-1);
}
int main(){
#ifdef ONLINE_JUDGE
#else
freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE
int N;
cin>>N;
for(int i = 0; i < N; i++)
scanf("%d", &pre[i]);
for(int i = 0; i < N; i++)
scanf("%d", &post[i]);
inOrder(0, 0, N-1);
printf("%s\n",flag==true?"Yes":"No");
for(int i=0;i<in.size();++i)
printf("%s%d",i>0?" ":"",in[i]);
return 0;
}
第二搏
???
柳神说过,出现莫名其妙的格式错误,就在末尾加\n;
我信了!