leetcode 594. 最长和谐子序列(python)