python 解决问题的集合-list index out of range