Shortest Path Problem Leetcode Python Easy Code