Implement Shortest Path Leetcode Python