欢迎访问宙启技术站
智能推送

Two-sum问题:Java函数实现

发布时间:2023-06-27 05:30:30

Java函数实现Two-sum问题

Two-sum问题是一道非常经典的算法问题,简而言之,就是在一个整数数组中,找出两个数的和为目标数。为了解决这个问题,可以使用一些较为常见的算法,例如暴力解法或者哈希表解法等。

在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");
}

在这个函数中,首先创建一个map对象,这个对象用来存储数组中每个元素所需要的补数。接着遍历数组,计算出每个元素需要的补数,如果在map中找到了这个补数,则表明已经找到了一组解答,直接返回这组解答的下标。如果没有找到,则将当前元素和下标保存到map中。

当程序运行结束之后,如果还没有找到解答,则抛出一个IllegalArgumentException异常。

为了验证这个函数的正确性,可以使用一些测试数据进行测试:

public static void main(String[] args) {
    int nums[] = {2, 7, 11, 15};
    int target = 9;
    int[] result = twoSum(nums, target);
    System.out.println(Arrays.toString(result));
}

输出结果:

[0, 1]

这个结果表明,数组中的第一个元素和第二个元素的和确实等于目标数。

总结

在Java中实现Two-sum算法问题,可以使用哈希表等常见算法。通过适当的测试,可以验证程序的正确性,确保程序能够正确的找到目标元素。