想问一下题目的输入输出要自己写吗，还是写一个函数就可以了？