백준 2263번 트리의 순회 C++

출처 : https://www.acmicpc.net/problem/2263

 

포스트오더는 루트를 마지막에 순회하는것을 이용해서 인오더에서 루트값을 찾아서 작은 트리로 분할해가면서 루트를 출력하는 방식으로 풀었다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long;

int n;

vector<int> inOrderIndex;
vector<int> postOrder;

void PrintRoot(int inOrderStart, int inOrderEnd, int postOrderStart, int postOrderEnd) {
	if (inOrderStart > inOrderEnd || postOrderStart > postOrderEnd)
		return;

	int root = postOrder[postOrderEnd];
	int rootIndex = inOrderIndex[root];
	int leftSize = rootIndex - inOrderStart;
	int rightSize = inOrderEnd - rootIndex;

	cout << root << " ";

	PrintRoot(inOrderStart, rootIndex - 1, postOrderStart, postOrderStart + leftSize - 1);
	PrintRoot(rootIndex + 1, inOrderEnd, postOrderEnd - rightSize, postOrderEnd - 1);
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n;
	inOrderIndex = vector<int>(n + 1);
	postOrder = vector<int>(n + 1);

	for (int i = 1; i <= n; i++) {
		int input;
		cin >> input;
		inOrderIndex[input] = i;
	}
	for (int i = 1; i <= n; i++) {
		int input;
		cin >> input;
		postOrder[i] = input;
	}
	PrintRoot(1, n, 1, n);
};