不用写输入吧，直接实现提供的函数接口，就像leetcode那样。