Leetcode 856 Python One Line