uu能贴一下第一题的python代码吗，想看看简洁的写法是什么样的