Python Leetcode 1004 Form