Shortest Path Problem Leetcode Python Beginner Code