更新参考代码

[newstart0514]

在原来的参考的代码，部分语句可能不太符合题目描述。