由中序序列和後序序列得到先序序列

上篇寫了個用先序序列和中序序列得到後序序列的題目

寫的有點繁瑣，這次來個簡潔點的。

思路都是一樣的

 

---

題目如下

 

題目描述

給出一棵二叉樹的中序與後序排列。求出它的先序排列。（約定樹結點用不同的大寫字母表示，長度\le 8≤8）。

輸入格式

22行，均爲大寫字母組成的字符串，表示一棵二叉樹的中序與後序排列。

輸出格式

11行，表示一棵二叉樹的先序。

輸入輸出樣例

輸入 #1複製

BADC
BDCA

輸出 #1複製

ABCD

---

---

 

直接上代碼：思路簡單不解釋

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

int N;
string pre,re,in;

void dfs(int reL,int reR,int inL,int inR) {
    if(reL>reR||inL>inR) return;
    char v = re[reR];
    putchar(v);
    int k;
    for(k=inL;k<=inR;k++) {
        if(in[k]==v) break;
    }
    int lenL = k-inL;
    dfs(reL,reL+lenL-1,inL,inL+lenL-1);
    dfs(reL+lenL,reR-1,k+1,inR);
}

int main(){
    cin >> in >> re;
    dfs(0,(int)re.length()-1,0,(int)in.length()-1);
}