Leetcode 1366 Python Solution