[Python] 인접 리스트(dictionary, set 활용) 구현

배코딩·2022년 1월 1일
0

note

목록 보기
7/151
import sys
from collections import deque
sys.setrecursionlimit(1000000)
input = sys.stdin.readline

N, M, V = map(int, input().split())
graph = {}

for i in range(M):
    a, b = map(int, input().split())

    if a not in graph:
        graph[a] = set([b])
    else:
        graph[a].add(b)

    if b not in graph:
        graph[b] = set([a])
    else:
        graph[b].add(a)

딕셔너리의 특정 키에 해당하는 밸류에 기존의 밸류는 놔두면서도 거기에 뭔가 추가하고싶으면,

처음에 밸류를 할당할 때 set으로 넣고, 이후에 어떤 값을 더 덧붙이고 싶으면 graph[key].add(추가로 덧붙일 값) 이런 식으로 set의 특성을 활용. value의 set에 직접 값을 추가해주면 됨

profile
PS, 풀스택, 앱 개발, 각종 프로젝트 내용 정리 (https://github.com/minsu-cnu)

0개의 댓글