题目:给出一个整型数列表和一个整数sum,求列表中加起来等于sum的两个数,并且这一对是在列表中最先组成对的。
这道题并不难,使用两个for循环很容易做出来。但提交答案时说出了错误:
Process was terminated. It took longer than 12000ms to complete
我在原来答案基础上做了少许更改,可始终达不到要求,无奈只好上网找寻答案。总结后思路如下:
- 首先出现问题的原因是在处理大列表时,双重for循环耗费太多资源,需要做的是把两个for精减为一个。
- 因为题目要求是返回最先组对成功的两个数,所以要找到列表中符合要求的数对中,第二个数最先出现的数对。
- 为了达到上一步骤的目的,把遍历过的数缓存起来,以后遍历的数在缓存好的数中查找是否有配对的,有则返回,无则继续,直到遍历完。
def sum_pairs(ints, s):
cache = set()
for i in ints:
other = s - i
if other in cache:
return [other, i]
cache.add(i)
修改后的比我那两个for简洁很多,看起来也清楚,时间更是降了很多,所以算法还是很重要的。
另外,缓存之所以用set,而不用list,是因为前者使用hash算法,查找速度为O(1),而后者需要通过下标遍历,查询越长的列表,需要的时间越长。dict与set拥有类似的结构。
所以,如果存储的数据会被反复查询,且量大,那么尽量不用list,而是使用dict或set。