n개의 정점을 갖는 이진 트리의 정점에 1부터 n까지의 번호가 중복 없이 매겨져 있다. 이와 같은 이진 트리의 인오더와 포스트오더가 주어졌을 때, 프리오더를 구하는 프로그램을 작성하시오.
첫째 줄에 n(1 ≤ n ≤ 100,000)이 주어진다. 다음 줄에는 인오더를 나타내는 n개의 자연수가 주어지고, 그 다음 줄에는 같은 식으로 포스트오더가 주어진다.
첫째 줄에 프리오더를 출력한다.
C++
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int inorder[100001], postorder[100001], index[100001];
void printPreOrder(int in_start, int in_end, int post_start, int post_end)
{
if (in_start > in_end || post_start > post_end) return;
const int root = postorder[post_end];
const int pos = index[root]; // 서브트리의 정점 위치 (inorder 기준)
const int L = pos - in_start;
cout << root << " ";
printPreOrder(in_start, pos - 1, post_start, post_start + L - 1);
printPreOrder(pos + 1, in_end, post_start + L, post_end - 1);
}
int main()
{
int n;
cin >> n;
for (int i = 1; i <= n; ++i)
{
cin >> inorder[i];
index[inorder[i]] = i;
}
for (int i = 1; i <= n; ++i)
cin >> postorder[i];
printPreOrder(1, n, 1, n);
}