分享一个第二题简洁的写法 python