题目:给出一个整型数列表和一个整数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。


对影yu
1 声望0 粉丝