💡문제
상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.
상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다.
출력
첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.
풀이💬
친구와, 친구의 친구를 구하면 되기 때문에 거리가 2이상 3이하인 노드의 개수를 구해주는 문제이다.
from collections import deque
n = int(input())
m = int(input())
graph = [[] for _ in range(n+1)]
def bfs(node):
queue = deque()
queue.append(node)
while queue:
node = queue.popleft()
for n in graph[node]:
if check[n] == 0:
check[n] = check[node]+1
queue.append(n)
for _ in range(m):
a, b = map(int, input().split())
graph[a].append(b)
graph[b].append(a)
check = [0] * (n+1)
check[1] = 1
bfs(1)
res = sum([1 for c in check if 2 <= c <= 3])
print(res)
기억하자.
✦DFS: 스택과 재귀함수로 구현할 것
✦BFS: deque, while로 구현, 시간복잡도가 낮은 popleft를 사용하자
graph를 표현할 때 인덱스 0번을 비우고 하면 더 편하다.
'알고리즘 문제' 카테고리의 다른 글
[Python]13549번.숨바꼭질3(그래프,BFS)⭐️ (0) | 2023.12.22 |
---|---|
[Python]16953번.A → B(그래프,BFS) (0) | 2023.12.21 |
[Python]2667번.단지붙이기(그래프,DFS,BFS) (1) | 2023.12.19 |
[Python]1697번.숨바꼭질(그래프이론, 그래프 탐색, BFS)⭐️ (0) | 2023.12.18 |
[Python]2606번.바이러스(그래프이론, 그래프 탐색, BFS, DFS) (1) | 2023.12.15 |