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

www.acmicpc.net/problem/11725

11725번: 트리의 부모 찾기

#include<iostream>
#include<vector>

using namespace std;
int n;
vector<int> info[100001];
bool checkArr[100001];
int result[100001];

void D(int _n)
{
	for (int i = 0; i < static_cast<signed>(info[_n].size()); i++)
	{
		int target = info[_n][i];
		if (checkArr[target] == false)
		{
			checkArr[target] = true;
			result[target] = _n;
			D(target);
		}
	}
}

int main()
{
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);
	cin >> n;
	int input1;
	int input2;

	//양방향 입력
	for (int i = 0; i < n - 1; i++)
	{
		cin >> input1 >> input2;
		info[input1].push_back(input2);
		info[input2].push_back(input1);
	}

	checkArr[1] = true;
	result[1] = 1;

	//재귀 시작
	D(1);

	for (int i = 2; i <= n; i++)
	{
		cout << result[i] << '\n';
	}

	return 0;
}