根据书中题目做的一个小程序