LeetCode 1. 两数之和

Posted 2022-12-02 阿乐246

一、题目详情

给定一个整数数组nums和一个整数目标值target，请你在该数组中找出和为目标值target的那两个整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例：

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

二、思路

首先能想到的就是，两层循环，遍历数组中的每一种组合。类似于概率论中的组合，从N个整数中选取两个，只要两数之和为target就可以了。这种方法当然可以，但是比较耗时，时间复杂度是O(N²)。
如果使用了HashMap将数组中每一个元素的索引 i 和值nums[i]都对应地存储起来，那么在第一层循环遍历每个元素的时候，只需要根据target-nums[i]查询HashMap就可以知道结果了，而Hashmap的查询时间是O(1)，总的时间复杂度是O(N)。

三、代码

class Solution 
    public int[] twoSum(int[] nums, int target) 
        Map<Integer,Integer> map=new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++) 
          int rest=target-nums[i];
          if(map.containsKey(rest)) 
            return new int[] i,map.get(rest);
          
          map.put(nums[i], i); //放在后面，防止同一个元素重复出现。
        
        throw new IllegalArgumentException("No two sum solution");