题目
定给一个整数数组nums
状语从句:一个目标值target
,请在你该数组中找出状语从句:为目标值的那两个整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定nums = [2,7,11,15],target = 9
因为nums [ 0 ] + nums [ 1 ] = 2 + 7 = 9
所以返回[ 0,1 ]
普通解法 时间O(n^2),空间O(1)
这可能是大家最快能想到的方法了,直接列出来(简单粗暴双重循环):
vector<int> twoSum(vector<int>& nums, int target) {
if (nums.empty())
{
return{};
}
for (int i = 0; i < nums.size(); ++i)
{
for (int j = i + 1; j < nums.size(); ++j)
{
if (nums[i] + nums[j] == target)
{
return {i, j};
}
}
}
return{};
}
LeetCode运行结果如下:
Hash表的空间换时间方式 时间O(n),空间O(n)
vector<int> twoSum(vector<int> &numbers, int target)
{
//map to index and vector value
//Key is the number and value is its index in the vector.
unordered_map<int, int> hash;
for (int i = 0; i < numbers.size(); i++) {
int res = target - numbers[i];
//原理就是:
/*
需要求a+b = 13 => 2 + 11 = 13
1. numbers[i] = 2 则 res = 11 此时11不存在于数组将此时的{2,0}记录
2. numbers[i] = 7 则 res = 6 此时6不存在于数组将此时的{7,1}记录
3. numbers[i] = 9 则 res = 4 此时4不存在于数组将此时的{9,2}记录
4. numbers[i] = 11 则 res = 2 此时2存在于数组所以hash.find(res) != hash.end()
res=2 通过hash[res]计算出容器的index; i = 3!
*/
if (hash.find(res) != hash.end()) {
return{ hash[res] ,i};
}
hash[numbers[i]] = i;
}
return{};
}
LeetCode运行结果如下:
最后给出一个相同的python版本,原理是想通的。
def twoSum(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
print(twoSum([2,7,9,11], 13))