Tree Recovery POJ - 2255 前中綴求後綴 DFS

題意：

已知二叉樹的前中序，求後序。

思路：

例如   PRE=DBACEGF IN=ABCDEFG 

首先，我們會把這棵樹分成三部分ROOT=PRE[0]、左子樹(BAC)，右子樹(EFG)，後序遍歷即爲左子樹、右子樹、根。

那麼我們再把左子樹、右子樹（若仍然存在）分解了，就可以得到整個樹的後序遍歷了。這是個遞歸的過程。

代碼：

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define LL long long
using namespace std;
const int maxn=20+10;
void dfs(string pre,string in)
{
    char c=pre[0];
    int len=in.size();
    int i=0;
    for(i=0;i<len;i++)
    {
        if(in[i]==c)break;
    }
    if(i>0)dfs(pre.substr(1,i),in.substr(0,i));//兩個參數是起始位置，長度
    if(i<len-1)dfs(pre.substr(i+1),in.substr(i+1));
    cout<<c;
}
int main()
{
    ios:
    string pre;
    string in;
    ios::sync_with_stdio(false);
    cin.tie(false);
    cin.tie(false);
    pre="12345";
    cout<<pre.substr(2,3);
    while(cin>>pre)
    {
        cin>>in;
        dfs(pre,in);
        cout<<endl;
    }
}