题目描述:
给定一个整数数组 nums
和一个目标值 target
,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
示例:
给定 nums = [2, 7, 11, 15], target = 9 因为 nums[0] + nums[1] = 2 + 7 = 9 所以返回 [0, 1]
暴力枚举(两重循环,O(n²))
C++写法:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
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 {};
}
};
执行用时: 204 ms, 在Two Sum的C++提交中击败了6.01% 的用户
内存消耗: 5.6 MB, 在Two Sum的C++提交中击败了0.82% 的用户
Python写法:
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
for i,n in enumerate(nums):
for j in range(i + 1, len(nums)):
if nums[i] + nums[j] ==target:
return i,j
执行用时: 3692 ms, 在Two Sum的Python提交中击败了42.61% 的用户
内存消耗: 7.8 MB, 在Two Sum的Python提交中击败了84.63% 的用户
提交为python3后,结果为:
执行用时: 44 ms, 在Two Sum的Python3提交中击败了99.77% 的用户
内存消耗: 7.9 MB, 在Two Sum的Python3提交中击败了46.97% 的用户
算法二:使用哈希函数:
C++实现:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> num2idx;
for(int i = 0;i<nums.size();++i){
if(num2idx.find(target -nums[i]) != num2idx.end()){
return {num2idx[target - nums[i]],i};
}
num2idx[nums[i]] = i;
}
return {};
}
};
执行用时: 24 ms, 在Two Sum的C++提交中击败了61.48% 的用户
内存消耗: 6.8 MB, 在Two Sum的C++提交中击败了0.82% 的用户
map是基于红黑树实现的,查找复杂度是log(n)
unordered_map内部使用一个大的hash_table,期望查找时间为O(1)。
Python3实现:
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
num2idx = dict()
for i,n in enumerate(nums):
if target - n in num2idx:
return num2idx[target - n], i
else:
num2idx[n] = i
执行用时: 44 ms, 在Two Sum的Python3提交中击败了99.77% 的用户
内存消耗: 7.9 MB, 在Two Sum的Python3提交中击败了46.97% 的用户