题目是这样的
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
自己初步的想法我相信多数人也应该想到了,就是两个for循环搞定。
public class TwoNumSum {
public int[] twoSum(int[] nums, int target) {
// int[] result = new int[2];
// for (int i = 0; i < nums.length ; i++) {
// for (int j = i + 1; j < nums.length ; j++) {
// if (nums[i] + nums[j] == target) {
// result[0] = nums[i];
// result[1] = nums[j];
// break;
// }
// }
// }
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 };
}
}
}
throw new IllegalArgumentException("No two solution");
}
}
下面是lettcode的两个更好的解法
public class TwoNumSum2 {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
map.put(nums[i], i);
}
for (int i = 0; i < nums.length; i++) {
int temp = target - nums[i];
System.out.println("temp:" + temp);
if (map.containsKey(temp) && (map.get(temp) != i)) {
return new int[] { i, map.get(temp) };
}
}
throw new IllegalArgumentException("No two sum solution");
}
}
还有一种
public class TwoNumSum3 {
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(complement)) {
return new int[] { map.get(complement), i };
}
map.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}