【思路】
这种题一般有二种形式,共同点是都已知中序序列。如果没有中序序列,是无法唯一确定一棵树的。
<1>已知二叉树的前序序列和中序序列,求解树。
1、确定树的根节点。树根是当前树中所有元素在前序遍历中最先出现的元素。
2、求解树的子树。找出根节点在中序遍历中的位置,根左边的所有元素就是左子树,根右边的所有元素就是右子树。若根节点左边或右边为空,则该方向子树为空;若根节点
边和右边都为空,则根节点已经为叶子节点。
3、递归求解树。将左子树和右子树分别看成一棵二叉树,重复1、2、3步,直到所有的节点完成定位。
<2>、已知二叉树的后序序列和中序序列,求解树。
1、确定树的根。树根是当前树中所有元素在后序遍历中最后出现的元素。
2、求解树的子树。找出根节点在中序遍历中的位置,根左边的所有元素就是左子树,根右边的所有元素就是右子树。若根节点左边或右边为空,则该方向子树为空;若根节点
边和右边都为空,则根节点已经为叶子节点。
3、递归求解树。将左子树和右子树分别看成一棵二叉树,重复1、2、3步,直到所有的节点完成定位。
测试用例:
【先序 中序 求 后序】
输入:
先序序列:ABCDEGF
中序序列:CBEGDFA
输出后序:CGEFDBA
代码:
- /*
- PreIndex: 前序序列字符串中子树的第一个节点在PreArray[]中的下标
- InIndex: 中序序列字符串中子树的第一个节点在InArray[]中的下标
- subTreeLen: 子树的字符串序列的长度
- PreArray: 先序序列数组
- InArray:中序序列数组
- */
- void PreInCreateTree(BiTree &T,int PreIndex,int InIndex,int subTreeLen){
- //subTreeLen < 0 子树为空
- if(subTreeLen <= 0){
- T = NULL;
- return;
- }
- else{
- T = (BiTree)malloc(sizeof(BiTNode));
- //创建根节点
- T->data = PreArray[PreIndex];
- //找到该节点在中序序列中的位置
- int index = strchr(InArray,PreArray[PreIndex]) - InArray;
- //左子树结点个数
- int LenF = index - InIndex;
- //创建左子树
- PreInCreateTree(T->lchild,PreIndex + 1,InIndex,LenF);
- //右子树结点个数(总结点 - 根节点 - 左子树结点)
- int LenR = subTreeLen - 1 - LenF;
- //创建右子树
- PreInCreateTree(T->rchild,PreIndex + LenF + 1,index + 1,LenR);
- }
- }
主函数调用:
- BiTree T;
- PreInCreateTree(T,0,0,strlen(InArray));
- PostOrder(T);
另一种算法:
- /*
- PreS 先序序列的第一个元素下标
- PreE 先序序列的最后一个元素下标
- InS 中序序列的第一个元素下标
- InE 先序序列的最后一个元素下标
- PreArray 先序序列数组
- InArray 中序序列数组
- */
- void PreInCreateTree(BiTree &T,int PreS ,int PreE ,int InS ,int InE){
- int RootIndex;
- //先序第一个字符
- T = (BiTree)malloc(sizeof(BiTNode));
- T->data = PreArray[PreS];
- //寻找该结点在中序序列中的位置
- for(int i = InS;i <= InE;i++){
- if(T->data == InArray[i]){
- RootIndex = i;
- break;
- }
- }
- //根结点的左子树不为空
- if(RootIndex != InS){
- //以根节点的左结点为根建树
- PreInCreateTree(T->lchild,PreS+1,(RootIndex-InS)+PreS,InS,RootIndex-1);
- }
- else{
- T->lchild = NULL;
- }
- //根结点的右子树不为空
- if(RootIndex != InE){
- //以根节点的右结点为根建树
- PreInCreateTree(T->rchild,PreS+1+(RootIndex-InS),PreE,RootIndex+1,InE);
- }
- else{
- T->rchild = NULL;
- }
- }
- PreInCreateTree(T,0,strlen(PreArray)-1,0,strlen(InArray)-1);
/*--------------------------------------- * 日期:2015-04-28 * 作者:SJF0115 * 题目: 105.Construct Binary Tree from Preorder and Inorder Traversal * 网址:https://leetcode.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/ * 结果:AC * 来源:LeetCode * 博客: -----------------------------------------*/ #include <iostream> #include <vector> #include <algorithm> using namespace std; struct TreeNode{ int val; TreeNode *left; TreeNode *right; TreeNode(int x):val(x),left(nullptr),right(nullptr){} }; class Solution { public: TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) { int size = preorder.size(); if(size <= 0){ return nullptr; }//if return PreInBuildTree(preorder,inorder,0,0,size); } private: TreeNode* PreInBuildTree(vector<int> &preorder,vector<int> &inorder,int preIndex,int inIndex,int size){ if(size <= 0){ return nullptr; }//if // 根节点 TreeNode* root = new TreeNode(preorder[preIndex]); // 寻找根节点在中序遍历数组的下标 int index = 0; for(int i = 0;i < size;++i){ if(preorder[preIndex] == inorder[inIndex+i]){ index = inIndex+i; break; }//if }//for // 左子树个数 int leftSize = index - inIndex; // 右子树个数 int rightSize = size - leftSize - 1; // 左子树 root->left = PreInBuildTree(preorder,inorder,preIndex+1,inIndex,leftSize); // 右子树 root->right = PreInBuildTree(preorder,inorder,preIndex+1+leftSize,index+1,rightSize); return root; } }; void PostOrder(TreeNode* root){ if(root){ PostOrder(root->left); PostOrder(root->right); cout<<root->val<<endl; }//if } int main(){ Solution solution; vector<int> preorder = {1,2,4,8,5,3,6,7}; vector<int> inorder = {8,4,2,5,1,6,3,7}; TreeNode* root = solution.buildTree(preorder,inorder); // 输出 PostOrder(root); return 0; }
具体讲解请看:点击打开链接
【中序 后序 求先序】
输入:
中序序列:CBEGDFA
后序序列:CGEFDBA
输出先序:ABCDEGF
代码:
- /*
- PostIndex: 后序序列字符串中子树的最后一个节点在PreArray[]中的下标
- InIndex: 中序序列字符串中子树的第一个节点在InArray[]中的下标
- subTreeLen: 子树的字符串序列的长度
- PostArray: 后序序列数组
- InArray:中序序列数组
- */
- void PostInCreateTree(BiTree &T,int PostIndex,int InIndex,int subTreeLen){
- //subTreeLen < 0 子树为空
- if(subTreeLen <= 0){
- T = NULL;
- return;
- }
- else{
- T = (BiTree)malloc(sizeof(BiTNode));
- //创建根节点
- T->data = PostArray[PostIndex];
- //找到该节点在中序序列中的位置
- int index = strchr(InArray,PostArray[PostIndex]) - InArray;
- //左子树结点个数
- int LenF = index - InIndex;
- //创建左子树
- PostInCreateTree(T->lchild,PostIndex - (subTreeLen - 1 - LenF) - 1,InIndex,LenF);
- //右子树结点个数(总结点 - 根节点 - 左子树结点)
- int LenR = subTreeLen - 1 - LenF;
- //创建右子树
- PostInCreateTree(T->rchild,PostIndex-1,index + 1,LenR);
- }
- }
主函数调用:
- BiTree T2;
- PostInCreateTree(T2,strlen(PostArray) - 1,0,strlen(InArray));
- PreOrder(T2);
/*--------------------------------------- * 日期:2015-05-01 * 作者:SJF0115 * 题目: 106.Construct Binary Tree from Inorder and Postorder Traversal * 网址:https://leetcode.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/ * 结果:AC * 来源:LeetCode * 博客: -----------------------------------------*/ #include <iostream> #include <vector> using namespace std; struct TreeNode{ int val; TreeNode *left; TreeNode *right; TreeNode(int x):val(x),left(nullptr),right(nullptr){} }; class Solution { public: TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) { int size = inorder.size(); if(size <= 0){ return nullptr; }//if return InPostBuildTree(inorder,postorder,0,size-1,size); } private: TreeNode* InPostBuildTree(vector<int> &inorder,vector<int> &postorder,int inIndex,int postIndex,int size){ if(size <= 0){ return nullptr; }//if // 根节点 TreeNode* root = new TreeNode(postorder[postIndex]); // 寻找postorder[postIndex]在中序序列中的下标 int index = 0; for(int i = 0;i < size;++i){ if(postorder[postIndex] == inorder[inIndex+i]){ index = inIndex+i; break; }//if }//for int leftSize = index - inIndex; int rightSize = size - leftSize - 1; root->left = InPostBuildTree(inorder,postorder,inIndex,postIndex-1-rightSize,leftSize); root->right = InPostBuildTree(inorder,postorder,index+1,postIndex-1,rightSize); return root; } }; void PreOrder(TreeNode* root){ if(root){ cout<<root->val<<endl; PreOrder(root->left); PreOrder(root->right); }//if } int main() { Solution solution; vector<int> inorder = {8,4,2,5,1,6,3,7}; vector<int> postorder = {8,4,5,2,6,7,3,1}; TreeNode* root = solution.buildTree(inorder,postorder); PreOrder(root); }
完整代码:
- #include<iostream>
- #include<string>
- using namespace std;
- //二叉树结点
- typedef struct BiTNode{
- //数据
- char data;
- //左右孩子指针
- struct BiTNode *lchild,*rchild;
- }BiTNode,*BiTree;
- //先序序列
- char PreArray[101] = "ABCDEGF";
- //中序序列
- char InArray[101] = "CBEGDFA";
- //后序序列
- char PostArray[101] = "CGEFDBA";
- /*
- PreIndex: 前序序列字符串中子树的第一个节点在PreArray[]中的下标
- InIndex: 中序序列字符串中子树的第一个节点在InArray[]中的下标
- subTreeLen: 子树的字符串序列的长度
- PreArray: 先序序列数组
- InArray:中序序列数组
- */
- void PreInCreateTree(BiTree &T,int PreIndex,int InIndex,int subTreeLen){
- //subTreeLen < 0 子树为空
- if(subTreeLen <= 0){
- T = NULL;
- return;
- }
- else{
- T = (BiTree)malloc(sizeof(BiTNode));
- //创建根节点
- T->data = PreArray[PreIndex];
- //找到该节点在中序序列中的位置
- int index = strchr(InArray,PreArray[PreIndex]) - InArray;
- //左子树结点个数
- int LenF = index - InIndex;
- //创建左子树
- PreInCreateTree(T->lchild,PreIndex + 1,InIndex,LenF);
- //右子树结点个数(总结点 - 根节点 - 左子树结点)
- int LenR = subTreeLen - 1 - LenF;
- //创建右子树
- PreInCreateTree(T->rchild,PreIndex + LenF + 1,index + 1,LenR);
- }
- }
- /*
- PostIndex: 后序序列字符串中子树的最后一个节点在PreArray[]中的下标
- InIndex: 中序序列字符串中子树的第一个节点在InArray[]中的下标
- subTreeLen: 子树的字符串序列的长度
- PostArray: 后序序列数组
- InArray:中序序列数组
- */
- void PostInCreateTree(BiTree &T,int PostIndex,int InIndex,int subTreeLen){
- //subTreeLen < 0 子树为空
- if(subTreeLen <= 0){
- T = NULL;
- return;
- }
- else{
- T = (BiTree)malloc(sizeof(BiTNode));
- //创建根节点
- T->data = PostArray[PostIndex];
- //找到该节点在中序序列中的位置
- int index = strchr(InArray,PostArray[PostIndex]) - InArray;
- //左子树结点个数
- int LenF = index - InIndex;
- //创建左子树
- PostInCreateTree(T->lchild,PostIndex - (subTreeLen - 1 - LenF) - 1,InIndex,LenF);
- //右子树结点个数(总结点 - 根节点 - 左子树结点)
- int LenR = subTreeLen - 1 - LenF;
- //创建右子树
- PostInCreateTree(T->rchild,PostIndex-1,index + 1,LenR);
- }
- }
- //先序遍历
- void PreOrder(BiTree T){
- if(T != NULL){
- //访问根节点
- printf("%c ",T->data);
- //访问左子结点
- PreOrder(T->lchild);
- //访问右子结点
- PreOrder(T->rchild);
- }
- }
- //后序遍历
- void PostOrder(BiTree T){
- if(T != NULL){
- //访问左子结点
- PostOrder(T->lchild);
- //访问右子结点
- PostOrder(T->rchild);
- //访问根节点
- printf("%c ",T->data);
- }
- }
- int main()
- {
- BiTree T;
- PreInCreateTree(T,0,0,strlen(InArray));
- PostOrder(T);
- printf("\n");
- BiTree T2;
- PostInCreateTree(T2,strlen(PostArray) - 1,0,strlen(InArray));
- PreOrder(T2);
- return 0;
- }