两数之和--力扣

最新推荐文章于 2024-09-09 18:27:53 发布

是珊子哇

最新推荐文章于 2024-09-09 18:27:53 发布

阅读量77

点赞数

分类专栏： Java算法文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_46598198/article/details/133744191

版权

Java算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

文章描述了一种算法，利用Java的HashMap实现，给定一个整数数组，寻找和为目标值的两个数及其下标。通过遍历数组，检查目标值与当前元素之差是否在哈希表中，找到就返回，否则将当前元素加入哈希表继续搜索。

摘要由CSDN通过智能技术生成

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

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

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

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]

示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

思路：使用哈希表来解决这个问题，遍历数组nums，对于每个元素num，首先判断target-num是否在哈希表中，如果在，直接返回两个数的下标，如果不在，则将它加入哈希表，以便之后判断。

import java.util.HashMap;

class Solution {
    public int[] twoSum(int[] nums, int target) {

        Scanner sc = new Scanner(System.in);
        HashMap<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.get(complement)表示获取complement所对应的下标。如果complement存在于哈希表中，则表示在前面的元素中有一个元素与当前元素的和为target，直接返回它们的下标；否则，将当前元素加入哈希表中，以便之后的判断。如果最终没有找到符合条件的两个元素，则抛出异常。

if (map.containsKey(complement)) {
    int index1 = map.get(complement);
    int index2 = i;
    int[] result = new int[]{index1, index2};
    return result;
} else {
    map.put(nums[i], i);
}

map.containsKey(key) 是一个HashMap类中的方法，用于检查HashMap中是否包含指定的键。

在给定的代码中，map.containsKey(complement) 判断哈希表 map 是否包含键 complement。如果存在，则说明前面已经有一个元素与当前元素的和为目标值 target，即找到了答案；否则，说明还没有找到对应的两个数，将当前元素加入哈希表中以便后续判断。