문제
n개의 정점을 갖는 이진 트리의 정점에 1부터 n까지의 번호가 중복 없이 매겨져 있다. 이와 같은 이진 트리의 인오더와 포스트오더가 주어졌을 때, 프리오더를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 n(1 ≤ n ≤ 100,000)이 주어진다. 다음 줄에는 인오더를 나타내는 n개의 자연수가 주어지고, 그 다음 줄에는 같은 식으로 포스트오더가 주어진다.
출력
첫째 줄에 프리오더를 출력한다.
예제 입력 1
3 1 2 3 1 3 2
예제 출력 1
2 1 3
#include <iostream>
using namespace std;
const int maxnum = 100001;
int postorder[maxnum];
int inorder[maxnum];
int index[maxnum];
void PrintPreorder(int istart, int iend, int pstart, int pend)
{
if (istart > iend || pstart > pend)
return;
int rootnum = index[postorder[pend]];
int left = rootnum - istart;
int right = iend - rootnum;
cout << inorder[rootnum] << " ";
PrintPreorder(istart, rootnum - 1, pstart, pstart + left - 1);
PrintPreorder(rootnum+1, iend, pstart+left, pend-1);
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
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);
return 0;
}
728x90
반응형
'알고리즘 > solved.ac' 카테고리의 다른 글
[class4] (백준 15686) 치킨 배달 (0) | 2022.02.25 |
---|---|
[class4] (백준 11444) 피보나치 수 6 (0) | 2022.02.24 |
[class4] (백준 1918) 후위 표기식 (0) | 2022.02.21 |
[class4] (백준 1865) 웜홀 (0) | 2022.02.19 |
[class4] (백준 1167) 트리의 지름 (0) | 2022.02.18 |