问题描述
在整数数组nums中找出两个和为target的下标
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
输入:nums = [3,2,4], target = 6
输出:[1,2]
枚举法
两层遍历,一一对比
class Solution {
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
}
}
return new int[]{-1,-1};
}
}
哈希法
一层遍历,但引入缓存,先判断当前元素是否被缓存,如果缓存及找到了,否则把当前元素的差值放入缓存。
相比枚举法省略了非法元素的相互判断
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>(nums.length);
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(target - nums[i])) {
return new int[]{map.get(target - nums[i]),i};
}
map.put(nums[i], i);
}
return new int[]{-1,-1};
}
}
并行法
以上解法的限制是每次遍历只能比较一个数值,如果一次能比较两个数值,速率必然加快。并行法是两层遍历,但是第一层取首位两个元素,内层循环每次取一个比较
class Solution {
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length / 2; i++) {
int l = nums.length - i - 1;
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
if (nums[l] + nums[j] == target) {
return new int[]{j, l};
}
}
}
return new int[]{-1, -1};
}
}
更进阶的是除了外层循环取首位两个元素,内层循环也取两个元素
class Solution {
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length / 2; i++) {
int l = nums.length - i - 1;
for (int j = i + 1; j < nums.length; j++) {
int k = l - 1;
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
if (nums[l] + nums[j] == target) {
return new int[]{j, l};
}
if (nums[i] + nums[k] == target) {
return new int[]{i, k};
}
if (nums[k] + nums[j] == target) {
return new int[]{j, k};
}
if (nums[k] + nums[l] == target) {
return new int[]{k, l};
}
}
}
return new int[]{-1, -1};
}
}