一道简单的两数之和问题(LeetCode)/java

最新推荐文章于 2025-12-20 22:59:04 发布

原创最新推荐文章于 2025-12-20 22:59:04 发布 · 258 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

文章介绍了如何利用哈希表解决LeetCode上的TwoSum问题。通过遍历数组一次，将元素和其索引存入哈希表，然后再次遍历数组查找目标值的差值，如果在哈希表中找到则返回对应下标，从而达到O(n)的时间复杂度。

封面/米涅巴顶楼

题目：

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出和为目标值 target 的那两个整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/two-sum

示例1：

示例2：

示例3：

这个问题是经典的 Two Sum 问题，可以使用哈希表来解决。首先遍历一遍数组，将每个数值和它对应的下标存入哈希表中。然后再遍历一遍数组，对于每个数值，计算目标值与该数值的差值，然后在哈希表中查找是否存在该差值，如果存在，则说明找到了一组解，返回对应的下标即可。

具体思路如下：

创建一个哈希表，用于存储数组中每个元素及其下标。
遍历数组，对于每个元素 nums[i]，计算出与其配对的数 target - nums[i]。
在哈希表中查找配对的数，如果存在，则返回它们的下标。
如果哈希表中不存在配对的数，则将当前元素添加到哈希表中。

功能代码实现如下 :

package test;


import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class Test4 {
    public static void main(String[] args) {
        int[] nums = {2, 7, 11, 15};
        int target = 9;
        Test4 test = new Test4();
        System.out.println(Arrays.toString( test.twoSum(nums,target)));
        //返回的数组中，第一个元素为第一个数的索引，第二个元素为第二个数的索引
    }

    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");
        //如果遍历整个数组都没有找到合适的结果，则抛出异常

    }
}

时间复杂度小于 O(n2) 的算法：get√

使用哈希表（HashMap）来实现时间复杂度小于 O(n2) 的算法。

这个方法的时间复杂度是 O(n)，空间复杂度也是 O(n)。