Leetcode 856 Python Example