Baekjoon
백준 5567 결혼식 / Python
Tcoding
2023. 3. 21. 12:03
728x90
728x90
https://www.acmicpc.net/problem/5567
5567번: 결혼식
예제 1의 경우 2와 3은 상근이의 친구이다. 또, 3과 4는 친구이기 때문에, 4는 상근이의 친구의 친구이다. 5와 6은 친구도 아니고, 친구의 친구도 아니다. 따라서 2, 3, 4 3명의 친구를 결혼식에 초대
www.acmicpc.net
코드
더보기
from collections import deque
def bfs(x):
q = deque()
visited[x] = 1
q.append(x)
while q:
now = q.popleft()
for i in arr[now]:
if visited[i] == 0:
q.append(i)
visited[i] = visited[now] + 1
n = int(input())
m = int(input())
arr = [[0] * (n + 1) for _ in range(n + 1)]
visited = [0 for _ in range(n + 1)]
for _ in range(m):
a, b = map(int, input().split())
arr[a].append(b)
arr[b].append(a)
bfs(1)
ans = 0
for i in range(2, n + 1):
if 1 <= visited[i] <= 3:
ans += 1
print(ans)
728x90
300x250