提示
给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 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 <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
- 只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2)
的算法吗?
枚举法(O(n2))
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
int* result = (int*)malloc(2 * sizeof(int));
*returnSize = 2;
for(int i=0;i<numsSize-1;i++)//蛮力法,一个个枚举出来进行相加比较
{
for(int j=i+1;j<numsSize;j++)//第一轮把nums[0]与所有数进行相加比较,以此类推每个数进行相加比较,直到出现和等于target的一对数并分别返回下标
if(nums[i]+nums[j]==target)
{
result[0]=i;
result[1]=j;
break;
}
}
return result;
}
哈希表法
用哈希表可以在常数时间内执行插入和查找操作,从而减少总体的时间复杂度。
这段代码的思路是:
- 遍历数组,对于每个元素,计算与目标值之间的差值。
- 利用哈希表存储已经遍历过的数字以及它们的下标。
- 检查当前数字的差值是否在哈希表中,如果在,则返回对应的下标。
这个方法的时间复杂度为O(n),因为哈希表的插入和查找操作都是 O(1),并且只需要遍历一次数组。
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
int* result = (int*)malloc(2 * sizeof(int));
*returnSize = 2;
// 创建一个哈希表,用于存储数字和对应的下标
int hash_table[100000] = {0}; // 假设数组中的元素范围在 -50000 到 49999 之间
for (int i = 0; i < numsSize; i++) {
// 计算当前数字与目标值之间的差值
int complement = target - nums[i];
// 查找哈希表中是否存在与差值相等的数
if (hash_table[complement + 50000] != 0) {
// 如果存在,则返回当前下标和哈希表中存储的下标
result[0] = hash_table[complement + 50000] - 1;
result[1] = i;
break;
}
// 如果不存在,则将当前数字及其下标存入哈希表中
hash_table[nums[i] + 50000] = i + 1;
}
return result;
}