From 2af13495d8b41bfea8c29e12bfdfa6f603d7418e Mon Sep 17 00:00:00 2001 From: seongwon030 <105052068+seongwon030@users.noreply.github.com> Date: Fri, 12 Apr 2024 20:47:11 +0900 Subject: [PATCH] =?UTF-8?q?2024-04-12=20=EA=B2=B0=ED=98=BC=EC=8B=9D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../\352\262\260\355\230\274\354\213\235.py" | 32 +++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 "seongwon030/bfs/\352\262\260\355\230\274\354\213\235.py" diff --git "a/seongwon030/bfs/\352\262\260\355\230\274\354\213\235.py" "b/seongwon030/bfs/\352\262\260\355\230\274\354\213\235.py" new file mode 100644 index 0000000..9779c02 --- /dev/null +++ "b/seongwon030/bfs/\352\262\260\355\230\274\354\213\235.py" @@ -0,0 +1,32 @@ +from collections import deque +import sys + +input = sys.stdin.readline + +n = int(input()) +m = int(input()) +arr = [[0 for i in range(n)] for i in range(n)] + +for i in range(m): + a,b = map(int,input().split()) + arr[a-1][b-1] = 1 # 노드 저장 + arr[b-1][a-1] = 1 +visited = [0 for i in range(len(arr))] + +def bfs(start): + q = deque() + q.append(start) + visited[start] = 1 + while q: + top = q.popleft() + for i in range(len(arr)): + if arr[top][i] == 1 and visited[i] == 0: + visited[i] = visited[top]+1 + q.append(i) + +bfs(0) +re = 0 +for i in visited: + if i==2 or i==3: + re+=1 +print(re) \ No newline at end of file