欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

哈希表简单算法题:在数组中找到两个数,它们之和等于给定的数字。

程序员文章站 2024-03-15 20:12:18
...

这段时间为了校招找工作开始刷LeetCode,第一道题就是这道题:
两数之和:在数组中找到两个数,它们之和等于给定的数字。
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
乍一看很简单,两个循环暴力**就行了。

class Solution {
    public int[] twoSum(int[] nums, int target) {
        for(int i=0;i<nums.length;i++){
            for(int j=i+1;j<nums.length;j++){
                if(nums[i]+nums[j]==target){
                    return new int[]{i,j};
                }
            }
        }
        throw new IllegalArgumentException("There is not such two integer!");
    }
}

但是这样做的话,时间空间复杂度都会比较高,看评论的时候发现了用哈希表的方法。哈希表我之前几乎没有用过,这道题正好给了一次机会去简单熟悉一下哈希表的用法。
哈希表是一堆由键和键值组成的数据对的集合,是以索引的方法进行查找。查找时根据键就可以查找到键值,其中键必须是唯一的,键值可以重复。这道题是需要返回所求的两个数的数组下标,所以我们可以以这数值为键,数组下标为键值,建立哈希表求解。代码如下

class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++){
            int leave=target-nums[i];
            if(map.containsKey(leave)){
                return new int[] {map.get(leave),i};
            }
            map.put(nums[i],i);
        }
        throw new IllegalArgumentException("There is not such two integer!");
    }
}

其中只用了一次for循环,在循环里实现了哈希表存储数据和哈希表的查询,时间空间复杂度就由之前的O(n²)变成了O(n)。
用到了哈希表的两个方法:containsKey()和get()。第一个是根据键判断哈希表中是否存在这个键,返回值是Boolean。第二个是根据键在哈希表中查找到对应的键值。