[BOJ] 결혼식

---

코드

import sys
# from collections import deque
input = sys.stdin.readline

n = int(input())
m = int(input())
adj = [[] for _ in range(n+1)]
inv = [0]*(n+1)
ans = 0

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

for friend in adj[1]:
    inv[friend] = 1
    for f in adj[friend]:
        inv[f] = 1

for i in range(n+1):
    if i == 1: continue
    if inv[i] == 1: ans += 1

print(ans)

결과

풀이 방법

• 인접리스트 adj에 각 동기의 친구들을 저장한다 (양방향)
• 1번(상근)의 각 친구들의 인접노드(친구)를 순회하면서 초대멤버로 등록하면 된다

---