5.1.4 Python解法