파이썬 그래프 관련 실수모음

실수1. (s, v, w)주어졌을 때 v->s고려 안함

배경

프로그래머스 : 합승 택시 요금
fares로 (시작노드, 도착노드, 가중치)의 리스트가 들어올때(ex. fares: [[4, 1, 10], [3, 5, 24], [5, 6, 2], [3, 1, 41], [5, 1, 24], [4, 6, 50], [2, 4, 66], [2, 3, 22], [1, 6, 25]]), 이렇게 그래프를 만들었다.

그러더니 KeyError가 난다...

뭐가 문제일까?

풀이

해당 문제는 fares가 무방향그래프라서 s->v 뿐만아니라 v->s도 고려해야하는데, 위 코드는 s->v만을 신경써서 그렇다.

따라서 그래프가 무방향인지 방향인지 꼭 먼저 따지는 습관을 가져야한다.

아래는 고친 코드의 예시이다.