【力扣高频题】1.两数之和 - 算法启航

最新推荐文章于 2024-07-25 11:44:32 发布

强连通子图

最新推荐文章于 2024-07-25 11:44:32 发布

阅读量622

点赞数 18

分类专栏：力扣高频题文章标签：算法 leetcode

本文链接：https://blog.csdn.net/weixin_46879314/article/details/139408993

版权

力扣高频题专栏收录该内容

10 篇文章 0 订阅

订阅专栏

最近有个刚毕业去面试的小伙伴私信我说，遇到了很新奇的面试方式：

一环扣一环，层层递进，一直在紧张状态下，还好准备的比较充分，回答的不错，希望也能尽快收到二面通知。

那正好，今天我们就来讲讲这道力扣入门简单题。

1. 两数之和

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

思路分析

其实，最简单、也是最容易想到的解法就是 暴力循环 。

思路也很简单，遍历数组中的每个元素，找是否存在另外一个元素与当前元素值之和为目标数，两层for循环搞定。

暴力循环

public int[] twoSum(int[] nums, int target) {
    int n = nums.length;
    for (int i = 0; i < n; i++) {
        // 从 i+1 号元素往后寻找
        for (int j = i + 1; j < n; j++) {
            if (nums[i] + nums[j] == target) {
                return new int[] { i, j };
            }
        }
    }
    return new int[0];
}

很显然，暴力循环的方法其时间复杂度为 $O(N^2)$ ，空间复杂度为 $O (1)$ 。

算法优化

思考一下，暴力循环 存在的不足是什么呢？

缺点：对数组中的元素存在 多次遍历 ，每来到一个新的元素，都要遍历一遍其后面所有的元素。存在大量重复行为！！！

因此，找到了问题的核心，自然而然就想出了解决办法。

如果能够在遍历一遍的同时，记录下之前遍历过的结果（即能够在 $O (1)$ 的复杂度下直接找到之前已经遍历过的元素下标）。就无需多次遍历，节省了时间。

而 哈希表 正是利用了空间换时间的思想，可以将查找时间从 $O (n)$ 降低到 $O (1)$ 。

我们在遍历数组元素时，将元素和其下标的对应关系记录在哈希表中，如果在已经遍历过的数组元素中存在与当前元素之和为 target 的元素下标（即，存在于哈希表中），直接返回答案。

优化代码

public static int[] twoSum(int[] nums, int target) {
    // key 某个之前的数 value 这个数出现的位置
    HashMap<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 中
        map.put(nums[i], i);
    }
    return new int[] { -1, -1 };
}