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;
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章