「优选算法刷题」:两数之和

45 阅读1分钟

一、题目

给定一个整数数组 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]

提示:

  • 2 <= nums.length <= 104
  • -109 <= nums[i] <= 109
  • -109 <= target <= 109
  • 只会存在一个有效答案

二、思路解析

这道题就适合用哈希表来解决,因为需要频繁地查找某一个元素。

因此,我们先遍历数组 nums,并使用 HashMap 来存储每个元素的值及其索引

而且哈希表中存放的是 nums 数组中的元素,那我们就需要去计算目标值与当前元素之差 diff = target - nums[i],然后检查 HashMap 中是否存在键值为 diff 的记录:

  • 如果存在,说明已找到符合条件的数对,返回它们的索引(当前元素的索引 i 和 HashMap 中 diff 对应的值)。
  • 如果不存在,则将当前元素的值及其索引存入 HashMap 中。

遍历完要还是没有返回,则说明没有找到和为 target 的元素之和,而力扣的编译器需要所有分支都有返回值,但我们仅在 if 语句的位置有返回值,所以还要在遍历结束后,返回一个默认值 [-1, -1],表示未找到。

其实这道题都说了,只会存在一个有效答案,但很无奈呀,力扣不让过,只能多加这一句了。

三、完整代码

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> hash = new HashMap<>();
        for(int i = 0; i < nums.length; i++){
            int x = target - nums[i];
            if(hash.containsKey(x)){
                return new int[]{i, hash.get(x)};
            }
            hash.put(nums[i], i);
        }
        return new int[]{-1, -1};
    }
}

以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!