用Java做LeetCode第一题：两数之和

Posted 2023-04-10 阿码德乌斯

题目：

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

此题主要考察HashMap的理解。两数相加等于一个数，已知结果数，求另一个数组中的x和y以及下标。如果不用HashMap的话，可以使用双层for循环实现

如下：

 此方法为暴力的解法，时间复杂度是O(n^2)。

所以更优的解决方案是使用Map，Map是一种key-value的存储结构，可以用key保存数值，用value再保存数值所在的下标。

再明确一下，这道题我们需要给出一个元素，判断这个元素是否出现过，如果出现过，返回这个元素的下标。

那么判断元素是否出现，这个元素就要作为key，所以数组中的元素作为key，有key对应的就是value，value用来存下标。

再遍历数组的时候，只需要向map去查询是否有和目前遍历元素匹配的数值，如果有，就找到匹配的对，如果没有，就把目前遍历的元素放进map中，因为map存放的就是我们访问过的元素。

如下：

 总结：

本题可以用来学习熟悉map是用来做什么的，以及map中的key和value分别用来存放什么，可以让我们更能清楚该什么时候用map，清楚它的好处，了解它的底层

LeetCode 力扣1. Two Sum 两数之和 Java 解法

LeetCode的第一题，英文单词书中 Abandon 一般的存在，让我们来看一下题目：

 

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].

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

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

 

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

 

从题目中我们可以得知此题必有答案可解，容易想到将数组中所有元素装进哈希表 map 中，又由于要返回的是数组下标，因此将数组元素作为哈希表的key，数组下标作为哈希表的value；

 

将数组元素装进 map 以后对数组所有元素再次使用for循环，用target目标值对每一个元素进行相减得到 t；如果此时 map 的key 中含有 t 并且 其value不等于当前的 i，就说明我们已经得到了所需的答案，此时只需返回 当前的 i 值和哈希表中对应 t 的 value即可。

 

代码实现：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap<Integer, Integer>();
        int[] ans = new int[2];

       //将数组元素装进哈希表 map 中
        for (int i = 0; i <= nums.length; i++){
            map.put(nums[i], i);
        }

        for(int i = 0; i <= nums.length; i++){
            int t = target - nums[i];
            //如果 map 中含有答案则返回其下标
            if (map.containsKey(t) && map.get(t) != i){
                ans[0] = i;
                ans[1] = map.get(t);
            }
        }
        return ans;
    }
}    

 

由于两次循环的条件是一样的，用两个for就会略显累赘，因而在这里可以把代码改的简洁一点：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap<Integer, Integer>();
        int[] ans = new int[2];

       //合并两个for循环
        for(int i = 0; i <= nums.length - 1; i++){
            int t = target - nums[i];
            if (map.containsKey(t) && map.get(t) != i){
                ans[0] = i;
                ans[1] = map.get(t);
            }
            map.put(nums[i], i);
        }
        return ans;
    }
}

 

 

LeetCode刷题是一个漫长的过程，笔者也不过刚刚开始；为了更好的学习故而写下自己的心得与体会.正所谓路漫漫其修远兮，吾将上下而求索。

 笔者水平有限，如果有什么错误还请不吝赐教！

 

参考资料：

https://leetcode.com/problems/two-sum/
https://leetcode-cn.com/problems/two-sum/