问题:给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
解法一:暴力解法
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int i,j;
for(i=0;i<nums.size()-1;i++)
{
for(j=i+1;j<nums.size();j++)
{
if(nums[i]+nums[j]==target)
{
return {i,j};
}
}
}
return {i,j};
};
};
// 定义一个函数twoSum,它接受一个整数指针nums(指向整数数组的起始位置),一个整数numsSize(表示数组的大小),
// 以及一个整数target(表示目标和),并返回一个整数指针result,该指针指向一个包含两个整数的数组,
// 这两个整数是数组中相加等于target的两个数的索引。
int* twoSum(int* nums, int numsSize, int target) {
int i,j; // 定义两个循环变量i和j,用于遍历数组。
int *result=NULL; // 定义一个整数指针result,并初始化为NULL。这个指针将用于存储找到的索引对。
// 外层循环,从数组的第一个元素开始遍历到倒数第二个元素。
for(i=0;i<numsSize-1;i++) {
// 内层循环,从外层循环当前元素的下一个元素开始遍历到数组的最后一个元素。
for(j=i+1;j<numsSize;j++) {
// 判断当前两个元素之和是否等于目标值target。
if(nums[i]+nums[j]==target) {
// 如果等于target,则动态分配内存来存储两个索引。
result=(int*)malloc(sizeof(int)*2);
// 将找到的索引赋值给result指向的数组。
result[0]=i;
result[1]=j;
// 返回结果指针。
return result;
}
}
}
// 如果没有找到满足条件的索引对,则返回NULL。
return result;
}
malloc给result分配了两个int类型大小的空间,可知result是一个数组。
最容易理解,但是他的时间复杂为O(n2),因为这是两个for循环。
解法二:使用哈希表
class Solution {
public:
// 定义一个函数twoSum,接受一个整数向量nums和一个整数target,返回一个整数向量
vector<int> twoSum(vector<int>& nums, int target) {
// 创建一个哈希表a,用于存储数组元素和它们的索引
map<int,int> a; // 建立hash表存放数组元素
// 初始化一个大小为2的整数向量b,用于存放结果,初始值设为-1
vector<int> b(2,-1); // 存放结果
// 遍历数组nums,将数组元素和对应的索引插入到哈希表a中
for(int i=0;i<nums.size();i++)
a.insert(map<int,int>::value_type(nums[i],i));
// 再次遍历数组nums
for(int i=0;i<nums.size();i++) {
// 判断是否存在target - nums[i]这个键,并且该键对应的值不等于当前索引i
// 这样做是为了确保找到的两个数不是同一个数
if(a.count(target-nums[i])>0 && (a[target-nums[i]]!=i))
// 如果找到,则将当前索引i和target - nums[i]对应的索引存入结果向量b中
// 并跳出循环
{
b[0]=i;
b[1]=a[target-nums[i]];
break;
}
}
// 返回结果向量b
return b;
};
};
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> a;
vector<int> b(2,-1);
for(int i=0;i<nums.size();i++) {
if(a.count(target-nums[i])>0) {
b[0]=a[target-nums[i]]; // 正确的应该是这个数的索引
b[1]=i; // 当前数的索引
break;
}
a[nums[i]]=i;
}
return b;
};
};
不同于数组,哈希表找数的时间复杂度为O(1),所以这里的时间复杂度应为O(n)