트리의 부모 찾기

https://www.acmicpc.net/problem/11725

11725번: 트리의 부모 찾기

visited 배열을 사용하며

dfs 돌리면 끝.

dfs(1)로 시작한다.

ans 배열에 트리 노드의 부모를 저장

 

#include <iostream>
#include <vector>
 
using namespace std;
 
vector<int> ans;
vector<bool> visited;
vector<vector<int>> tree;
 
void dfs(int num)
{
    visited[num] = true;
 
    for (int now : tree[num])
    {
        if (!visited[now])
        {
            // 부모 노드를 정답 배열에 저장
            ans[now] = num;
            dfs(now);
        }
    }
}
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    int n;
    cin >> n;
    visited.resize(n + 1);
    tree.resize(n + 1);
    ans.resize(n + 1);
 
    for (int i = 1; i < n; i++)
    {
        int a, b;
        cin >> a >> b;
        tree[a].push_back(b);
        tree[b].push_back(a);
    }
 
    // 루트 노드에서 dfs
    dfs(1);
 
    for (int i = 2; i <= n; i++)
    {
        cout << ans[i] << '\n';
    }
}