알고리즘2 [Python] defaultdict와 인접리스트 Python으로 알고리즘 문제 풀이 중 인접리스트를 사용해야 풀리는 문제를 많이 볼 수 있었다. 인접리스트를 구현하는데 defaultdict를 사용하는 풀이가 자주 보여서 정리하고자 한다. collections.defaultdict - collections.defaultdict는 값(value)에 초깃값을 지정하여 딕셔너리를 생성하는 모듈 - 아래는 딕셔너리를 사용한 집계코드 # 딕셔너리를 사용한 집계코드 d = dict() for c in text: if c not in d: d[c] = 0 d[c] += 1 print(d) - 딕셔너리로 이와같은 집계코드를 작성할 땐, 중간의 if문 처럼 초깃값을 신경써야 한다. - 하지만, collections의 defaultdict를 사용하면 이러한 번거로움을 피.. 2023. 4. 14. [Python] PGS 부대복귀 성공 from collections import deque def solution(n, roads, sources, destination): answer = [] graph = [[] for _ in range(500001)] # 경로 정보를 담는 그래프 for src, des in roads: graph[src].append(des) graph[des].append(src) # 목적지에서 각 지역까지의 최단 경로를 구하는 BFS q = deque() q.append(destination) visited = [0 for _ in range(500001)] visited[destination] = 1 while q: now = q.popleft() for nq in graph[now]: if not vis.. 2023. 4. 13. 이전 1 다음