题目描述
给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
思路及解答
暴力求解(双指针法)( O(n^2) )
/*
* 思路:双指针法
*
*/
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] a = new int[2];
for(int i = 0; i < nums.length-1; i++){
for(int j = i+1; j < nums.length; j++){
if((target == nums[i] + nums[j]) && i != j){
a[0] = i;
a[1] = j;
}
}
}
return a;
}
}
哈希表两边循环( O(n) )
/*
* 思路:使用哈希表
* 第一次迭代中,将每个元素的值和它的索引添加到表中
*/
class Solution {
public int[] twoSum(int[] nums, int target) {
//创建哈希表
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
//第一次遍历,将元素添加到表中
for(int i = 0; i < nums.length; i++)
map.put(nums[i], i);
//第二次遍历,检查是否存在目标元素
for(int i = 0; i < nums.length; i++){
int complement = target - nums[i];
if(map.containsKey(complement) && map.get(complement) != i)
return new int[]{i, map.get(complement)};
}
throw new IllegalArgumentException("No two sum solution");
}
}
哈希表一遍循环( O(n) )
public static int[] twoSum(int[] nums, int target) {
//建立一个哈希表
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for(int i = 0; i < nums.length; i++){
// map.put(nums[i], i); (这句代码的位置有问题)
int complement = target - nums[i];
if(map.containsKey(complement) && map.get(complement) != i)
return new int[]{i, map.get(complement)};
map.put(nums[i], i); //(这句代码应该放到这里)
}
return null;
}
如果两个数是一样的这种情况,会出现问题。