提出input函数，消除重复代码