简单题,没有时间复杂度要求,所以当然用最简单的办法做啦(好孩子不要学我)
1.最简单的遍历法,因为题目假设每个输入只有一种答案,所以只需要遍历数组并检查nums[i]是否和目标值减去nums[j]相等就可以了,如果相等就输出当前i,j的值;
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[j] == target - nums[i]) {
return new int[] { i, j };
}
}
}
throw new IllegalArgumentException("No two sum solution");
}
leetcode的题解还给出了另外两种更好的方法:
2.两遍哈希表
基本思想是迭代两次数组,第一次迭代时构建哈希表,将每个元素的值和它的索引添加到表中,第二次迭代检查每个元素对应的目标元素(目标值-nums[i])是否存在于表中,目标元素不能是nums[i]本身;
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 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");
}
3.一遍哈希表
基本思想是在迭代时检查表中是否已经存在当前元素对应的目标元素,存在则可以直接返回;
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");
}