今天做了一道二叉树的题------------------美国血统 American Heritage
这道题我采用find函数查找root根节点,用substr提取字符段,然后用递归的方式查找到最深的根节点,并按照后序遍历的顺序输出。
#include<iostream>
using namespace std;
void dfs(string s1, string s2) {
char root = s2[0];//将前序遍历的首个字符(即根节点)拿出
int l = s1.find(root), r = s1.size() - 1 - l;//在中序遍历字符串中查找根的位置,从而得出左边子树长度,以及右边子树长度。
if (l) {//当左子树存在时进行下述操作
string s1_l = s1.substr(0, 1), s2_l = s2.substr(1, l);//进行字符串的提取,提取左子树出来,并递归进一步对左子树进行分割提取。
dfs(s1_l, s2_l);//递归
}
if (r) {
string s1_r = s1.substr(l + 1, r), s2_r = s2.substr(l + 1, r);//对右子树的操作同左子树
dfs(s1_r, s2_r);//递归
}
cout<<root;//对分割出的每个根节点进行输出
}
int main(){
string s1, s2;
cin >> s1 >> s2;//输入中序遍历字符串,以及前序遍历字符串。
dfs(s1, s2);
return 0;
}