Leetcode热题100 | Pro1 [1. 两数之和]

144 阅读1分钟

链接:leetcode.cn/problems/tw…

题目

给定一个整数数组 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
  • 只会存在一个有效答案

进阶: 你可以想出一个时间复杂度小于 O(n2) 的算法吗?

代码

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

代码解读:

这段代码使用哈希表data来解决两数之和的问题。

首先创建一个空哈希表。

然后遍历数组nums,对于每个数字nums[i],检查哈希表中是否存在能与之相加得到目标值target的数字。如果有,就返回这两个数字的索引;如果没有,就把当前数字及其索引存入哈希表,以便后续查找。

整个过程利用哈希表的快速查找特性,将时间复杂度降低到了 O(n)。