第一题可以用字典吧，会减少一个for循环