PAT A1020ツリー全検索[バイナリツリートラバーサル]

タイトル説明

リンク
層シーケンスの結果を求めて所定の配列と注文後、

分析

• 再帰、追求するためには、シーケンスと最初の注文後、本明細書に提示
• 再帰：プロセスの途中を考えてみましょう
  • プレオーダー\（[PREL、PRER] \ ） 配列中\（[INL、INR] \ ）
  • ルート：\（PREL \）
  • オーダールートに：\（K、A [K] == A [PRELは] \）得られたトラバース
  • 左サブツリー配列内のノードの数：\（K-INL \） 、左の部分木のシーケンス\（[INL、K-1 ] \）
  • 配列プレオーダー左サブツリー\（[PREL + 1、PREL + K-INL] \）
  • ノードの数の配列右サブツリー：\（INR-K \） 、右サブツリーシーケンス\（[K + 1、INR ] \）
  • 配列プレオーダー右サブツリー：\（[K + PREL-INL + 1、PRER] \）
  • 状態関数\（（PREL、PRER、INL 、INR）\）
• ターミネーター：0以下の配列の長さ、すなわち\（PRER-PREL <0 \ ）
• 具体的な動作内部：各再帰的には、それをハングアップするために、具体的方法を、あなたはルートノードを生成することができ、ルートノードの値を取得しますか？あなたがハングアップすることができたときに、関数の戻り値は、当時のポインタとして書かれたとすることができます。最後のリターンは、ツリー全体のルートであります

#include<bits/stdc++.h>
using namespace std;

struct node{
    int data;
    node *lchild;
    node *rchild;
};

const int maxn = 100;
int post[maxn], in[maxn];
node *create(int postL, int postR, int inL, int inR){
    if(postL > postR) return NULL;
    int k;
    for(k=inL;k<=inR;k++){
        if(in[k] == post[postR]) break;
    }
    int cnt = k-inL;
    node *root = new node;
    root->data = post[postR];
    root->lchild = create(postL, postL+cnt-1, inL, k-1);
    root->rchild = create(postL+cnt, postR-1, k+1, inR);
    return root;
}

vector<int> ans;
void layerOrder(node *root){
    queue<node*> q;
    q.push(root);
    while(!q.empty()){
        node *now = q.front();
        ans.push_back(now->data);
        q.pop();
        if(now->lchild) q.push(now->lchild);
        if(now->rchild) q.push(now->rchild);
    }
}

int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>post[i];
    }
    for(int i=0;i<n;i++){
        cin>>in[i];
    }
    node *root = create(0, n-1, 0, n-1);
    layerOrder(root);
    for(int i=0;i<ans.size();i++){
        if(i==0) cout<<ans[i];
        else cout<<" "<<ans[i];
    }
    cout<<endl;

}