【问题描述】
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
【解决示例】
对于给定的整数数组nums = [2, 7, 11, 15], 目标值target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以,最终结果需要返回 [0, 1]。
【问题分析】
首先在python中,整数数组可以使用python中的列表表示;由于我们需要找出两个整数的下标位置,所以当目标值已知时,可以使用目标值减去数组中的一个值num[a],记录其下标位置index_a;然后我们可以得到另外一个值b,在整个列表中查找等于b的值,并记录其下标位置index_b;最终返回[index_a, index_b]。
【问题解决】
按照我们上述的流程,可以写出如下函数:
使用一个新的列表list1来存储需要用到的索引,随后点击提交代码:
可以看到我们的代码是可以通过测试的,但是执行的时间与内存消耗都比较大,这是什么原因呢?我们可以看到在我们的代码中,使用了两层循环,这样会使得整个算法的时间复杂度大大增加,目前的时间复杂度为O(n^2)。
【评论区参考】
鉴于我们的代码的时间复杂度较高,我们参考一下评论区的代码:
参考了评论区大神的解法,他们都通过哈希来求解,使用字典来模拟哈希查询的过程。这样写的好处是,仅仅需要一层循环就可以搞定整个过程。
【代码优化】
从大神的代码中,可以进一步优化我们的算法,只需要判断目标值target减去nums[ i ]的值是否在列表中即可,如果在列表中,那么就直接使用list.index(key)的方法返回该值的索引;如果这个值不再列表中,那么可以选择返回空值或者不返回任何东西。根据这样的思路,我们将写出优化后的代码,优化后的代码如下所示:
我们尝试运行这段代码,可以发现还是通过的状态:
可以看到执行的用时已经大幅度提高了,但是内存消耗依然比较大。
【个人感悟】
代码是需要不断进行优化设计的,当一段代码能够解决问题时,我们还需要考虑的是如何降低算法的复杂度,如何减小运行的内存,最终不断完善自己的代码,达到最佳状态。