public int[] twoSum(int[] nums, int target) { Map<Integer, Integer> map = new HashMap<Integer, Integer>(); 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); } return new int[] {}; }使用该代码优化完成
时间: 2024-03-09 11:47:03 浏览: 125
Two Sum算法调试小demo
这个代码已经很优化了,不过我们可以稍微做一些改进。以下是优化后的代码:
```java
public static 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");
}
```
在这个代码中,我们将返回一个抛出异常的语句替换了原来的返回空数组的语句。这样做的好处是,如果没有找到符合条件的结果,程序会立即停止并抛出异常,避免了后续程序可能会出现的空指针异常等问题。
希望这个优化后的代码可以更好地帮助您解决问题!
阅读全文