1935. 二叉树重建
Description
对于二叉树T,可以递归定义它的先序遍历、中序遍历和后序遍历如下: PreOrder(T)=T的根节点+PreOrder(T的左子树)+PreOrder(T的右子树) InOrder(T)=InOrder(T的左子树)+T的根节点+InOrder(T的右子树) PostOrder(T)=PostOrder(T的左子树)+PostOrder(T的右子树)+T的根节点 其中加号表示字符串连接运算。例如,对下图所示的二叉树,先序遍历为DBACEGF,中序遍历为ABCDEFG。
输入一棵二叉树的先序遍历序列和中序遍历序列,输出它的广度优先遍历序列。
Input
第一行为一个整数t(0<t<10),表示测试用例个数。 以下t行,每行输入一个测试用例,包含两个字符序列s1和s2,其中s1为一棵二叉树的先序遍历序列,s2为中序遍历序列。s1和s2之间用一个空格分隔。序列只包含大写字母,并且每个字母最多只会出现一次。
Output
为每个测试用例单独一行输出广度优先遍历序列。
Sample Input
2 DBACEGF ABCDEFG BCAD CBAD
Sample Output
DBEACGF BCAD
先序遍历的第一个字符就是根,因此只需在中序遍历中找到它,就知道左右子树的先序和中序遍历了。例如:
先序:D BACEGF 中序:ABCD EFG
由先序遍历的第一个字符D 可知,此二叉树的根节点为D ,并且:
左子树的先序遍历为:BAC 中序遍历为:ABC
右子树的先序遍历为:EGF 中序遍历为:EFG
根据此规律,可通过递归来重构二叉树,然后层序遍历输出。
#include<iostream> #include<queue> #include<string> using namespace std; string pre,mid; int len,pos; struct Node{ Node *left; Node *right; char item; Node(char ch=0) { left=right=NULL; item=ch; } }; void build(Node *&root,int begin,int end) { if(begin>end||pos>=pre.size()) return ;//递归停止条件 root = new Node(pre[pos]);; int mpos=mid.find(pre[pos++]); //由前序遍历的首字母可以在中序遍历序列中查找对应位置,将中序的序列切开左子树跟右子树 build(root->left,begin,mpos-1);//递归构建 build(root->right,mpos+1,end); } int main() { int time; cin>>time; while(time--) { cin>>pre>>mid; len=pre.size(); pos=0; Node *root=new Node(); build(root,0,len-1); //根据输入的前序遍历跟中序遍历,构建一棵树 queue<Node*>q; q.push(root); while(!q.empty()) //层序遍历输出 { Node *midd=q.front(); q.pop(); cout<<midd->item; if(midd->left) q.push(midd->left); if(midd->right) q.push(midd->right); } cout<<endl; } return 0; }
(程序实现参考来自网络)自己实现还是有问题,树果然没学好。o(︶︿︶)o唉