[BOJ] 11725 - 트리의 부모 찾기

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

BOJ - 트리의 부모 찾기

문제 유형: 트리, bfs

문제 난이도: Silver II

시간 제한: 1초

메모리 제한: 256MB

 

문제

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

 

입력

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

 

출력

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

 

풀이

연결 관계만 주어지므로, 정확한 계층 관계를 알 수는 없다.

그래서, bfs를 하면서 방문처리를 하며 부모자식 관계를 지어주어야 한다.

1부터 bfs를 시작하면서, 현재 노드를 curr, 다음 인접합 노드들 중 하나를 next라 했을 때,

인접한 노드들 중에서 방문하지 않았다면 parent[next] = curr을 넣고, 방문처리하고 큐에 넣는다.

 

parent[2]부터 출력해준다.

 

코드

C++

#include <bits/stdc++.h>

using namespace std;

int main(int argc, char *argv[]) {
  int n;
  cin >> n;
  vector<vector<int>> graph(n + 1);
  int src, dst;
  for (int i = 0; i < n - 1; ++i) {
    cin >> src >> dst;
    graph[src].push_back(dst);
    graph[dst].push_back(src);
  }

  queue<int> q;
  vector<int> parent(n + 1);
  vector<bool> visited(n + 1, false);

  q.push(1);
  visited[1] = true;
  while (!q.empty()) {
    int curr = q.front();
    q.pop();

    for (const auto &next : graph[curr]) {
      if (visited[next])
        continue;
      parent[next] = curr;
      visited[next] = true;
      q.push(next);
    }
  }

  for (int i = 2; i < n + 1; ++i) {
    cout << parent[i] << "\n";
  }
  return 0;
}