温馨提示×

java如何实现两数之和

小亿
81
2024-12-07 08:26:33
栏目: 编程语言

在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中。如果遍历完整个数组都没有找到解决方案,我们抛出一个异常。

0