洛谷1030求先序排列(DFS)

题目内容:
给出一棵树的中序和后序排列，输出先序排列；
样例：中序：BADC
后序：BDCA
输出：ABCD

思路:根据树的遍历知识，后序的最后一个节点都代表一棵子树的根，如本例A为主根，然后从中序中遍历寻找主根A的位置，从而划分出左右子树；根据左右子树结点的个数，将后序划分，继续按之前操作步骤，递归搜索，先输出跟，再找左右子树，直到空树为止。

代码：
#include<bits/stdc++.h>
using namespace std;
string s1,s2;
void dfs(int l1,int r1,int l2,int r2)
{
if(l1>r1) return ; //所有节点都已经搜索完毕
cout<<s2[r2]; //输出根节点
int p=l1;
while(s1[p]!=s2[r2]) p++; //寻找根节点在中序所处的位置；
int ans=p-l1; //ans为左子树的节点个数
dfs(l1,p-1,l2,l2+ans-1); //遍历左子树；因为下标从0开始，所以-1;
dfs(p+1,r1,l2+ans,r2-1);
}

int main()
{
cin>>s1>>s2; //s1,s2分别为中序和后序；
int len=s1.length()-1;
dfs(0,len,0,len); //前两个是搜索中序，后两个是搜索后序;
return 0;
}