题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
2 <= nums.length <= 103
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
分析
方法一 暴力求解
双层循环,成立条件i != j && nums[i] + nums[j] == target
创建一个用于存储记录下标的数组。
C:
int* twoSum(int* nums, int numsSize, int target, int* returnSize){
int *tmp = (int*)malloc(2 * sizeof(int));//创建存放返回值的空间
for(int i = 0; i < numsSize; i++)
{
for(int j = i + 1; j < numsSize; j++)
{
if(nums[i] + nums[j] == target)
{
tmp[0] = i;
tmp[1] = j;
*returnSize = 2;//返回2个
return tmp;
}
}
}
*returnSize = 0;//返回0个
return tmp;
}
C++:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> tmp;
for(int i = 0; i < nums.size(); i++)
{
for(int j = i + 1; j < nums.size(); j++)
{
if(nums[i] + nums[j] == target)
{
tmp.push_back(i);
tmp.push_back(j);
return tmp;
}
}
}
return tmp;
}
};
JAVA:
class Solution {
public int[] twoSum(int[] nums, int target) {
int len = nums.length;
for(int i = 0; i < len; i++)
{
for(int j = i + 1; j < len; j++)
{
if(nums[i] + nums[j] == target)
{
return new int[]{i, j};//创建匿名数组
}
}
}
return new int[0];//返回空数组
}
}
复杂度分析
时间复杂度:O(N^2),其中 N 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。
空间复杂度:O(1)。
方法二 哈希表查找
用哈希表<值,下标>来倒着存放。
如果不考虑有俩相同的元素,我们可以用map或者unordered_map这种不重复的map容器。
先构造哈希表,然后一次遍历查找就可以了。如下:
C++:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> hashmap;//提供一对一的hash
vector<int> tmp;//存放返回值
for(int i = 0; i < nums.size(); i++)//构造哈希表
{
hashmap.push_back(pair<int, int>(nums[i], i));
//反过来放入map中,用来获取结果下标
}
for(int i = 0; i < nums.size(); i++)//查找
{
if(a.count(target-nums[i]) > 0)
{//nums[i]还未存入哈希表中,如果表中有另一半直接跳出
b.push_back(a[target - nums[i]]);
b.push_back(i);
break;
}
}
return b;
};
};
但是碰到用例输入:nums = [3,3], target = 6 输出:[0,1]
就会出现问题,因为map不允许有重复的key值元素,所以我们可以先判断然后根据结果来确定是返回,还是存放入哈希表中。
C++:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> hashmap;//提供一对一的hash
vector<int> tmp;//存放返回值
for(int i = 0; i < nums.size(); i++)//查找
{
if(hashmap.count(target-nums[i]) > 0)
{//nums[i]还未存入哈希表中,如果表中有另一半直接跳出
tmp.push_back(hashmap[target - nums[i]]);
tmp.push_back(i);
break;
}
hashmap.insert(pair<int, int>(nums[i], i));
//反过来放入map中,用来获取结果下标
}
return tmp;
};
};
JAVA:
class Solution {
public int[] twoSum(int[] nums, int target) {
int len = nums.length;
HashMap<Integer,Integer> hashmap = new HashMap<>();
for(int i = 0; i < len; i++)//查找
{
if(hashmap.containsKey(target-nums[i]))
{//nums[i]还未存入哈希表中,如果表中有另一半直接跳出
return new int[]{i,hashmap.get(target-nums[i])};
}
hashmap.put(nums[i], i);
//反过来放入map中,用来获取结果下标
}
return new int[0];
}
}
本章完