给定一个整型数组,找出能相加起来等于一个特定目标数字的两个数。
函数 twoSum 返回这两个相加起来等于目标值的数字的索引,且 index1 必须小于 index2。 请记住你返回的答案(包括 index1 和 index2)都不是从 0 开始的。
你可以假定每个输入都有且仅有一个解决方案。
输入: numbers={2, 7, 11, 15}, target=9
输出: index1=1, index2=2
这是一道非常经典的题目,暴力方法破解时间复杂度为O(n^2),逐一相加与target进行比较。该方法特性是通俗易懂,但是时间复杂度较高。
public static int[] twoSum1(int[] nums, int target) {
// O(n^2)
int index[] = {0,0};
for(int i = 0; i < nums.length; i++){
for(int j = i + 1; j < nums.length; j++){
if(nums[i] + nums[j] == target){
index[0] = i;
index[1] = j;
return index;
}
}
}
return index;
}
优化的方法比较容易想到的就是哈希MAP,时间复杂度为O(n),同时空间复杂度也是O(n),是典型的用空间换时间的方法,代码如下:
public static int[] twoSum(int[] nums, int target) {
// O(n)
int len = nums.length;
HashMap<Integer, Integer> hm = new HashMap<Integer, Integer>();
for(int i = 0; i < len; i++){
hm.put(nums[i], i);
}
int index[] = {0,0};
for(int i = 0; i < len; i++){
int complement = target - nums[i];
if(hm.containsKey(complement) && hm.get(complement) != i){
index[0] = i;
index[1] = hm.get(complement);
return index;
}
}
return index;
}
c++代码:
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int, int> map;
int len = nums.size();
for (int i = 0; i < len; i++){
if (map.find(target - nums[i]) != map.end()) {
return {map[target - i], i};
}
map[nums[i]] = i;
}
return {-1, -1};
}
};
int main() {
vector<int> nums = {3,2,4};
Solution sl;
vector<int> result = sl.twoSum(nums, 6);
cout << result[0] << result[1] << endl;
return 0;
}