[백준] 5567번 결혼식 - Python

백준 실버2 5567번 결혼식 - Python

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

5567번: 결혼식

---

문제

---

코드

import sys
C = int(sys.stdin.readline().rstrip())
N = int(sys.stdin.readline().rstrip())
graph = [[] for i in range(C+1)]
for _ in range(N):
    com1, com2 = map(int, sys.stdin.readline().rstrip().split())
    graph[com1] += [com2]
    graph[com2] += [com1]
ans = list()
for i in graph[1]:
    ans.append(i)
    for j in graph[i]:
        ans.append(j)
ans = list(set(ans))
l = len(ans)
if l == 0:
    print(0)
else:
    print(l-1)

---

후기

저번에 풀었던 바이러스 문제랑 거의 비슷한 것 같아서 복붙하려고 했는데 좀 다르더라고요

고치다가 변수명 빼고 완전 달라져버렸네요