题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路
可能很多人首先想到的是遍历,然后再遍历一次, 看有没有等于 target - array[i], 这样时间复杂度是O(n^2).
我们考虑空间换时间, 用hash map来处理。
代码
#include<iostream>
#include<unordered_map>
using namespace std;
void twoSum(int* nums, int numsSize, int target, int* returnSize){
unordered_map <int, int> m;
int i = 0;
for (i = 0; i < numsSize; ++i) {
if ((m.find(target - nums[i]) != m.end()) && m[target - nums[i]] != i){
returnSize[0] = i;
returnSize[1] = m[target - nums[i]];
}
m[nums[i]] = i;
}
return ;
}
int main() {
int array[5] = {2, 7, 13, 5, 8};
int tempArray[2];
twoSum(array, 5, 20, tempArray);
cout << tempArray[0] << " " << tempArray[1] << endl;
return 0;
}