修改第一题中的输出函数