깊이/너비 우선 탐색(DFS/BFS) | Find-UNION
🌏 문제
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
👽 풀이 ➡️ find-union 이용
def solution(n, computers):
def find_parent(parent, a):
if parent[a] != a:
parent[a] = find_parent(parent, parent[a])
return parent[a]
def union_parent(parent, a, b):
a = find_parent(parent, a)
b = find_parent(parent, b)
if a < b:
parent[b] = a
else:
parent[a] = b
parent = [i for i in range(n)]
for i in range(n):
for j in range(i, n):
if computers[i][j] == 1:
union_parent(parent, i, j)
for i in range(n):
parent[i] = find_parent(parent, i)
return len(set(parent))
👽 풀이 ➡️ BFS 이용
from collections import deque
def solution(n, computers):
answer = 0
visited = [False for _ in range(n)]
def bfs(com):
visited[com] = True
queue = deque([])
queue.append(com)
while queue:
com = queue.popleft()
visited[com] = True
for connect in range(n):
if connect != com and computers[com][connect] == 1:
if visited[connect] == False:
queue.append(connect)
for com in range(n):
if visited[com] == False:
bfs(com)
answer += 1
return answer
728x90
'💡코딩테스트 > 프로그래머스' 카테고리의 다른 글
[프로그래머스 | 파이썬 python] 단어 변환 (0) | 2023.05.17 |
---|---|
[프로그래머스 | 파이썬 python] 게임 맵 최단거리 (0) | 2023.05.17 |
[프로그래머스 | 파이썬 python] 타겟 넘버 (0) | 2023.05.16 |
[프로그래머스 | 파이썬 python] 등굣길 (0) | 2023.05.16 |
[프로그래머스 | 파이썬 python] 정수 삼각형 (0) | 2023.05.16 |