java笔试题：用python实现一个函数