题目
给定一个整数数组
nums
和一个目标值target
,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [3,2,3] target= 6
因为 nums[0] + nums[2] = 3 + 3 = 6
所以返回 [0, 2]
解析:
常规解法是通过暴力法,即将数组做循环嵌套进行两个整数相加即可行到结果,但此种解法得到的时间复杂度为O(n*n)。想要优化时间复杂度为O(n),需要将循环的层数降低为一层,此时就需要再借助其他的存储空间,即是用空间换时间。
暴力解法O(n*n):
vector<int> twoSum(vector<int>& nums, int target) {
for(int i = 0; i < nums.size() -1; i ++) {
for(int j = i + 1; j < nums.size(); j ++) {
if(nums[i] + nums[j] == target) {
return vector{i, j};
}
}
return vector<int>();
}
优化解法O(n)
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> _map;
for(int i = nums.size() - 1; i >=0 ; i --) {
int temp = target - nums[i];
if(_map.find(temp) == _map.end()) {
_map[nums[i]] = i;
continue;
}
return vector{i, _map[temp]};
}
return vector<int>();
}