给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
这个题暴力搜索就不分析了,说一下思路就行,暴力搜索就是设置两个循环,逐个查找一个数与其它所有数的和是否有目标数。
最好的方法就是使用哈希表(这个哈希表就是个字典吧!)
算法思路如下:
将目标数减去当前数target-nums[n]放到哈希表中作为键,其值为n。然后查找表中是否有下一个数,如果有,那么第一个数的索引就是查找到的那个差的索引,第二个数的索引就是下一个数的索引。
C++源代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
map<int, int> temp; //键为target-num[n],值为n
for(int i=0;i<nums.size();++i)
{
if(temp.count(nums[i]) != 0)
{
result.push_back(temp[nums[i]]);
result.push_back(i);
}
temp[target-nums[i]] = i;
}
return result;
}
};
python3源代码如下:
class Solution:
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
temp = {}
for i in range(len(nums)):
if temp.__contains__(nums[i]):
return [temp[nums[i]], i]
temp.update({target-nums[i]: i})
return -1