1. 题目
题目链接:https://leetcode-cn.com/problems/two-sum/
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
2. 解题
1. 暴力解法
每个元素都跟后面的比较,n2 复杂度
class Solution
{
public:
vector<int> twoSum(vector<int>& nums, int target)
{
int i,j;
vector<int> ans;
for(i = 0; i < nums.size()-1; ++i)
{
for(j = i+1; j < nums.size(); ++j)
{
if(nums[i]+nums[j] == target)
{
ans.push_back(i);
ans.push_back(j);
return ans;
}
}
}
return ans;
}
};
2. 哈希法
利用unordered_map哈希查找,先建立哈希表O(n),再查找O(n)
//哈希法
class Solution
{
public:
vector<int> twoSum(vector<int>& nums, int target)
{
int i, d;
vector<int> ans;
unordered_map<int,int> unomap;//unordered_map是哈希表
for(i = 0; i < nums.size(); ++i)//建立哈希表
unomap[nums[i]] = i;//哈希数组下标是值,存的原来的下标
for(i = 0; i < nums.size(); ++i)
{
d = target - nums[i];
if(unomap.find(d) != unomap.end() && unomap[d]>i)
{//如果查找到了,且不是同一个元素
ans.push_back(i);
ans.push_back(unomap[d]);
break;
}
}
return ans;
}
};
- 优化,只需要一次遍历
class Solution { // 2020.10.3
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> m;
int number;
for(int i = 0; i < nums.size(); ++i)
{
number = target-nums[i];
if(m.find(number) != m.end())
return {i, m[number]};
m[nums[i]] = i;
}
return {};
}
};
3. python3解答
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
dic = {}
for i in range(len(nums)):
dic[nums[i]] = i
for i in range(len(nums)):
if (target-nums[i]) in dic and dic[target-nums[i]] > i:
return [i,dic[target-nums[i]]]
return [-1,-1]
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!