LeetCode
题目:Two Sum
语言:Python
题目描述:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
例子:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
这个题O(n2)没法解,而且数组并没有排过序没有什么特点。
因为本题需要将元素和索引对应起来,最好的将两者结合起来的是什么数据结构呢?
Hash Table
在一个list中通过Index方法查找元素,需要O(n2)。但是如果是在hash table中,查找元素,最好O(1),最差O(n),但平均下来是O(1)。
在Python中直接使用dict,创建一个字典即可。
下面程序的逻辑是:
1.创建一个字典。
2.对于nums中所有的元素,检查是否在字典中,如果在字典中返回对应的索引和自己的索引组成的列表。
3.如果不在字典中,则将与自己相加等于target的数值与自己的索引添加到字典。
class Solution:
def twoSum(self, nums, target):
dict = {}
for i in range(len(nums)):
if nums[i] in dict:
return [dict[nums[i]], i]
else:
dict[target-nums[i]] = i
时间和空间复杂度都是O(n)