考研复习数据结构时写的代码