写了一下编程题的python版本