在Java中,要实现两数之和的问题,可以通过使用HashMap来存储已经遍历过的数字及其索引。这样,在遍历数组的过程中,可以快速检查是否存在一个数与当前数的和等于目标值。以下是一个示例代码:
import java.util.HashMap;
import java.util.Map;
public class TwoSum {
public static void main(String[] args) {
int[] nums = {2, 7, 11, 15};
int target = 9;
int[] result = twoSum(nums, target);
System.out.println("两数之和为: [" + result[0] + ", " + result[1] + "]");
}
public static int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> numMap = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (numMap.containsKey(complement)) {
return new int[] {numMap.get(complement), i};
}
numMap.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
}
}
在这个示例中,我们首先定义了一个名为twoSum
的方法,该方法接受一个整数数组和一个目标值作为参数。我们使用一个HashMap来存储数组中的每个元素及其索引。然后,我们遍历数组,对于每个元素,我们计算目标值与该元素的差值(称为补数),并检查HashMap中是否已经存在这个补数。如果存在,说明我们找到了两个数,它们的和等于目标值,我们返回这两个数的索引。如果不存在,我们将当前元素及其索引添加到HashMap中。如果遍历完整个数组都没有找到解决方案,我们抛出一个异常。