UVA536 二叉树重建 Tree Recovery

简介: UVA536 二叉树重建 Tree Recovery

题意

输入一棵二叉树的先序遍历和中序遍历序列,输出它的后序遍历序列。

输入

DBACEGF ABCDEFG
BCAD CBAD

输出

ACBFGED
CDAB

思路:我们可以通过二叉树的先序和中序可以推出后序节点.先序得根,中序分左右,通过递归便可输出后序节点.

此题中只是输出后序,没有必要存储起来

#include<bits/stdc++.h>
using namespace std;
string preorder,inorder;
void solve(string pre,string in){
  int index = in.find(pre[0]);
  if(pre==""){//递归结束条件 
    return;
  }
  solve(pre.substr(1,index),in.substr(0,index)) ;
  solve(pre.substr(index+1),in.substr(index+1));
  cout<<pre[0];
}
int main()
{
  while(cin>>preorder>>inorder){
    solve(preorder,inorder);
    cout<<endl;
  }
  return 0;
}

由于上面是参数传递的是字符串,中间用到了substr函数,在处理时效率较低,所以我们可以把参数改为传递索引,这样效率就大大提高了.

参考代码2

#include<bits/stdc++.h>
using namespace std;
string preorder,inorder;
void solve(int pre,int in,int len){
  if(len <= 0){//结束条件 
    return;
  }
  int index = inorder.find(preorder[pre])-in;//根节点在中序的位置索引 -- 字符串起始索引就是根节点左右子树的结点数. 
  solve(pre+1,in,index);
  solve(pre+index+1,in+index+1,len-index-1);//左右子树的节点数不一定相等,所以得用总长度-左子树结点的个数  要注意索引是从0开始的.所以最后要-1 
  cout<<preorder[pre]; 
} 
int main()
{
  while(cin>>preorder>>inorder){
    int len = preorder.size();
    solve(0,0,len);//分别指代 先序起始索引,中序起始索引,  长度 
    cout<<endl;
  }
  return 0;
}

注:substr(index,pos),从index开始截取pos个字符.如果第二个参数不写,默认截取到最后.

相关文章
|
11月前
|
人工智能
Tree Recovery
Tree Recovery
34 0
|
机器学习/深度学习 存储 C++
【PAT甲级 - C++题解】1123 Is It a Complete AVL Tree
【PAT甲级 - C++题解】1123 Is It a Complete AVL Tree
78 0
UVA122 树的层次遍历 Trees on the level
UVA122 树的层次遍历 Trees on the level
【1110】Complete Binary Tree (25分)【完全二叉树】
【1110】Complete Binary Tree (25分)【完全二叉树】 【1110】Complete Binary Tree (25分)【完全二叉树】
113 0