HihoCoder - 1049 後序遍歷

描述

在參與過了美食節之後,小Hi和小Ho在別的地方又玩耍了一陣子,在這個過程中,小Ho得到了一個非常有意思的玩具——一棵由小球和木棍連接起來的二叉樹

小Ho對這棵二叉樹愛不釋手,於是給它的每一個節點都標記了一個標號——一個屬於A..Z的大寫字母,並且沒有任意兩個節點的標號是一樣的。小Hi也瞅準了這個機會,重新鞏固了一下小Ho關於二叉樹遍歷的基礎知識~就這樣,日子安穩的過了兩天。

這天,小Ho正好在求解這棵二叉樹的前序、中序和後序遍歷的結果,但是卻在求出前序遍歷和中序遍歷之後不小心把二叉樹摔到了地上,小球和木棍等零件散落了一地!

小Ho損失了心愛的玩具,正要嚎啕大哭起來,所幸被小Hi發現了,勸說道:“彆着急,這不是零件都還在麼?拼起來不就是了?”

“可是我忘記了二叉樹長什麼樣子了!”小Ho沮喪道。

“這個簡單,你不是剛剛求出了這棵二叉樹的前序和中序遍歷的結果麼,利用這兩個信息就可以還原出整棵二叉樹來哦!”

“這樣麼?!!”小Ho止住了淚水,問道:“那要怎麼做呢?”

沒錯!小Ho在這一週遇到的問題便是:給出一棵二叉樹的前序和中序遍歷的結果,還原這棵二叉樹並輸出其後序遍歷的結果。

提示:分而治之——化大爲小,化小爲無

輸入

每個測試點(輸入文件)有且僅有一組測試數據。

每組測試數據的第一行爲一個由大寫英文字母組成的字符串,表示該二叉樹的前序遍歷的結果。

每組測試數據的第二行爲一個由大寫英文字母組成的字符串,表示該二叉樹的中序遍歷的結果。

對於100%的數據,滿足二叉樹的節點數小於等於26。

輸出

對於每組測試數據,輸出一個由大寫英文字母組成的字符串,表示還原出的二叉樹的後序遍歷的結果。

Sample Input
AB
BA
Sample Output
BA

解題思路:這道題就是題目名的表達意思,知道前序,中序,求後序

代碼:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
struct node
{
    node *l,*r;
    char data;
};
node *build(char *pre,char *mid,int n)
{
    for(int i=0;i<n;i++)
    {
        if(pre[0]==mid[i])
        {
            node *s = new node();
            s->data = mid[i];
            s->l = build(pre+1,mid,i);
            s->r = build(pre+i+1,mid+i+1,n-i-1);
            return s;
        }
    }
    return NULL;
}
void travel(node *t)
{
    if(!t)
        return;
    travel(t->l);
    travel(t->r);
    printf("%c",t->data);
}
int main()
{
    char mid[50],pre[50];
    while(cin>>pre>>mid)
    {
        node *t = new node();
        t = build(pre,mid,strlen(mid));
        travel(t);
        cout<<endl;
    }
    return 0;
}


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