[백준, python] 5567번 - 결혼식

백준 문제풀이 시작

---

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다.

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

---

풀이

풀이 확인하기

해당 문제는 1을 root로 하는 dfs 문제로 바꿔 볼 수 있다.

입력을 받은 뒤 리스트로 변환하고, 1번째 리스트를 제외한 나머지 리스트들 안에서 1이 보일 경우 1과 연결되어있단 소리이므로 리스트를 초기화해준 뒤 해당 그래프를 dfs를 돌려 결과를 얻는다.

import sys
input = sys.stdin.readline

vertex = int(input())
edge = int(input())

graph = [[] for _ in range(vertex + 1)]

for _ in range(edge):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

for i in graph:
    i.sort()

for i in range(2, len(graph)):
  is_1 = False
  for j in range(0, len(graph[i])):
    if(graph[i][j] == 1):
      is_1 = True
      break
  if not is_1:
      graph[i] = []

a = []
visited = [0] * (vertex + 1)

num = 0
def dfs(start):
  global num
  visited[start] = 1
  num += 1
  print(start, end=" ")
  for i in graph[start]:
    if not visited[i]:
      dfs(i)

dfs(1)
print(num-1)

• baekjoon
• python