python solution leetcode 226