[Python] 파이썬 Dictionary를 이용해 그래프 만들기

문제는 다음과 같다.

 

여러 줄에 걸쳐 서로 연결된 노드의 관계들을 입력받는다고 하자.

 

# 입력
5    # 서로 연결된 관계의 수
1 2
2 5
5 1
3 4
4 6

 

다음과 같이 입력받았을 때, 파이썬의 딕셔너리와 리스트를 이용해 그래프를 표현할 것이다.

 

n = int(input())
dic = {}

for i in range(n):
	a, b = map(int, input().split())
	dic[a] = dic.get(a, []) + [b]
	dic[b] = dic.get(b, []) + [a]
    
print(dic)

 

여기서 딕셔너리의 get함수를 모른다면 여기를 참고하자😉

 

암튼, 저렇게 딕셔너리를 채우고 출력해보면

 

{1: [2, 5], 2: [1, 5], 5: [2, 1], 3: [4], 4: [3, 6], 6: [4]}

 

다음과 같이 그래프가 만들어진다🤗