学习《算法》4的代码