Shortest Path Problem Leetcode Python Beginner