建议

不是 瞧你们忙的 这种题如果是为了纯刷题可以不写函数 在主函数判断就行
若是练习 建议就按题目要求写函数