本题用自定义函数怎么做