LeetCode刷题日: 两数之和

85 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第20天,点击查看活动详情

前言

周三周三,努力工作!每日一到简单题提升自信心。

[1. 两数之和] 难度:简单,选了一道leetcode开篇一题。

题目

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

审题

这题有两种解法,第一种是暴力解法,非常简单两个次for循环nums数组对nums数组进行target - nums[i]然后去第二个nums数组每一个都寻找一遍直到等于target - nums[i]这个值便得出答案,这个方法固然可行,但是时间复杂度来到了O(n)2 这个时间是非常耗时的,虽然题目的长度并不长没有超时。所以得另寻门道才行,想要时间下去那么必然要牺牲空间,这两个是死对头你要这个你就得丢掉另一个。我们空间选取的是HashMap哈希表来做,原因有2,1.hash表中的api可以快速帮我定位target - nums[i],也就是containsKey这个api返回是告诉你这个值是否存在 2. 我们可以通过get方法快速得到hash表中的数据,无需我们去遍历。

所以我给出的代码是一次循环完成,然后通过matchMap.containsKey(lastNum)去map中寻找剩余数,没有的话就往map中存放下次循环再次寻找。

我的解题代码

    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> matchMap = new HashMap<>();

        int[] result = new int[2];

        for (int i = 0; i < nums.length; i++) {
            int lastNum = target - nums[i];
            if (matchMap.containsKey(lastNum)) {
                result[0] = i;
                result[1] = matchMap.get(lastNum);
                break;
            }
            matchMap.put(nums[i], i);
        }
        return result;
    }

耗时

image.png