不用读取输入，直接写函数，main函数里面把测试用例写好调用，我一般都这么干