java小白刷题日记——两数之和

66 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第22天,点击查看活动详情

两数之和

难度 简单15981

给定一个整数数组 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) 的算法吗?

通过次数4,058,738提交次数7,681,708

请问您在哪类招聘中遇到此题?

开始还是我自己的解法,简单粗暴

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] ans=new int[2];
        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<nums.length;j++){
                if(nums[i]+nums[j]==target){
                    ans[0]=i;
                    ans[1]=j;
                }
            }
        }
        return ans;
    }
}

但是看了官方的解法之后,我发现我这简单粗暴的代码依然有可以优化的地方,比如可以省去ans数组,直接return new int[]{i, j};,然后在最后return new int[0];就可以了,不过感觉这两种写法区别不太大。

然后我们就来到了第二种高大上的解法了,哈希表.

那我们又来学新的java语法了

哈希表

  • 散列表,存储的内容是键值对(key-value)映射。

  • 无序,即不会记录插入的顺序。

  • 继承于AbstractMap,实现了 Map、Cloneable、java.io.Serializable 接口。

  • HashMap 的 key 与 value 类型可以相同也可以不同,可以是 字符串(String) 类型的 key 和 value,也可以是 整型(Integer) 的 key 和字符串(String)类型的 value。

  • 创建 HashMap 对象 Sites HashMap<String,String> Sites = new HashMap<String, String>();

  • 添加键值对 Sites.put("one", "Google");   

  • 用 get(key) 方法来获取 key 对应的 value System.out.println(Sites.get(3));

  • 删除对应键值对 Sites.remove(4);

  • 删除所有键值对(key-value) Sites.clear();

  • 计算HashMap 中的元素数量 Sites.size()

  • 迭代HashMap

// 输出 key 和 value  
        for (Integer i : Sites.keySet()) {  
            System.out.println("key: " + i + " value: " + Sites.get(i));  
        }  
        // 返回所有 value 值  
        for(String value: Sites.values()) {  
          // 输出每一个value  
          System.out.print(value + ", ");  
        }

还有更多的方法大家就自行查阅文档吧,这里只列出了很常用的几个方法。

那么熟悉了关于哈希表的方法,就可以来做这道题了,先建立一个哈希表,然后遍历数组,同时将键值对存入哈希表,在遍历到后边的数时,如果发现target-nums[i]的值能在哈希表中找到,那么就可以直接返回这两个数的key。

以下是这种解法的代码

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

image.png 可以看到,这种解法省时也省内存。

刷题第二天,每天进步一点点