[백준 C++] 11725. 트리의 부모 찾기

문제 링크

문제

루트 없는 트리가 주어진다. 이때, 트리의 루트를 1이라고 정했을 때, 각 노드의 부모를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 노드의 개수 N (2 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N-1개의 줄에 트리 상에서 연결된 두 정점이 주어진다.

출력

첫째 줄부터 N-1개의 줄에 각 노드의 부모 노드 번호를 2번 노드부터 순서대로 출력한다.

예제 입력 / 출력

// 예제 입력 1
7
1 6
6 3
3 5
4 1
2 4
4 7

// 예제 출력 1
4
6
1
3
1
4

// 예제 입력 2
12
1 2
1 3
2 4
3 5
3 6
4 7
4 8
5 9
5 10
6 11
6 12

// 예제 출력 2
1
1
2
3
3
4
4
5
5
6
6

풀이

📍 알고리즘

• 양방향으로 그래프를 벡터에 나타내고, DFS 알고리즘을 이용하여 부모 트리를 찾아 배열에 저장해서 부모 트리를 한 번에 출력한다.

#include <iostream>
#include <vector>

using namespace std;

const int MAX = 100001;


bool visited[MAX];
int parentNode[MAX];
vector<int> node[MAX];

void findParentNode(int start) {

	visited[start] = true;

	for (int i = 0; i < node[start].size(); i++) {

		int next = node[start][i];

		if (visited[next] == false) {
			parentNode[next] = start;
			findParentNode(next);
		}


	}
}

int main(void) {

	int N, temp1, temp2;

	cin >> N;

	for (int i = 0; i < N - 1; i++) {
		cin >> temp1 >> temp2;

		node[temp1].push_back(temp2);
		node[temp2].push_back(temp1);
	}

	findParentNode(1);

	for (int i = 2; i <= N; i++) cout << parentNode[i] << "\n";
}