原题网址(中):https://leetcode-cn.com/problems/two-sum/
原题网址(英):https://leetcode.com/problems/two-sum/
题目:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的两个整数。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路:
这题最简单的思路就是直接暴力求解,每读取一个数据便查看数组中其他数据与之和为target对应值。但是时间复杂度高,为
O
(
n
2
)
O(n^2)
O(n2)。
这里推荐采用哈希法完成,牺牲空间来换取时间。在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。此时时间复杂度降到
O
(
n
)
O(n)
O(n).
代码:
class Solution {
public static int[] twoSum(int[] nums, int target) {
int[] index = new int[2];//存储数组下标
HashMap<Integer, Integer> hashMap = new HashMap<>();
int length = nums.length;
for(int i = 0; i < length; i++) {
if(hashMap.containsKey(nums[i])) {
index[0] = hashMap.get(nums[i]);
index[1] = i;
return index;
//可以简写为 return int[]{hashMap.get(nums[i]), i};
}else {
hashMap.put(target-nums[i], i);
}
}
return index;
}
}