每日一道 LeetCode (1):两数之和

引言

前段时间看到一篇刷 LeetCode 的文章,感触很深,我本身自己上大学的时候,没怎么研究过算法这一方面,导致自己直到现在算法都不咋地。

一直有心想填补下自己的这个短板,实际上又一直给自己找理由各种推脱。

结果今天晚上吃多了,下定决心要刷一波 LeetCode 的题,刷题的过程顺便写点文章分享下(其实有很好的督促作用)。

LeetCode 的背景啥的我就不多介绍了,我计划只刷简单难度和中等难度的题,据说刷了这两个难度基本上就够了,至于困难这个难度,先等我把前两个部分刷完再说。

大致聊一下刷题的套路,先看题目, 5 分钟左右没思路的直接看答案,这玩意没思路是真没办法,没有基础储备想做题还是有点困难的。

看完答案自己动手写一下代码,这一点很重要,现在面试很多白板代码,一定要自己写,写会有效加深记忆力。

这个系列的文章计划日更,原本以为不是一件很难的事情,结果当我看到了这个:

1700+ 多道题,即使刨除掉困难的部分, 2/3 也有 1000+ 道题,我真的是给自己开了一个非常棒的头,这一下把未来两三年的规划都制定好了,我真是哔了狗了。

不过不管怎么样吧,决心都下好了,那么做还是要做的,对我这个计划感兴趣的同学可以每天在留言区和我一起打卡,预计每篇文章阅读时长在 3 分钟左右,写代码加调试代码总时长不会超过半小时。使用的代码为 Java ,如果使用 Python 写的话有点太取巧了。

做事情,重要的是要坚持。

需要代码朋友可以访问我的 GitHub 和 Gitee 获取,每天的代码我会同步提交至这两个代码仓库:

GitHub:https://github.com/meteor1993/LeetCode

Gitee:https://gitee.com/inwsy/LeetCode

题目:两数之和

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

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

示例:

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

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

思路

首先整理下思路,虽然题目的目标是要求一个加法,使用程序直接写加法有点不是那么好写,稍微转变一下,使用目标值 target 减去数组 nums 中的一个值,如果得到的结果也在这个数组中,那么我们就解题完成。

暴力破解

我最先想到的也是最简单的方案,就是暴力破解,直接两个循环套一下,暴力去算,得到结果:

public int[] twoSum_1(int[] nums, int target) {
    for (int i = 0; i < nums.length; i++) {
        int temp = target - nums[i];
        for (int j = i + 1; j < nums.length; j++) {
            if (nums[j] == temp) {
                return new int[]{i, j};
            }
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}

这种方案的缺点是时间复杂度有点高,优点是简单,应该是每个人都能想到的方案。

因为总共有 n 个元素,而对于其中的每个元素来讲,我们都需要遍历整个数组来寻找是否存在它所需要的对应的元素,这将耗费 O(n) 的时间。

时间复杂度: \(O(n^2)\)

空间复杂度: O(1) 。

两次哈希表

对暴力破解方案的优化思路是,整个数组,至少需要迭代一次,关键是在这次迭代中,我们要寻找另一种方案,能比套一层循环更快更高效的找到检查在整个数组中,是否含有我们需要的值。

我们可以借助哈希表来进行寻找,它支持以 「近似」 恒定的时间进行快速查找。

public int[] twoSum_2(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        map.put(nums[i], i);
    }
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement) && map.get(complement) != i) {
            return new int[] {i, map.get(complement)};
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}

我们把包含有 n 个元素的列表遍历两次。由于哈希表将查找时间缩短到 O(1) ,所以时间复杂度为 O(n)。

时间复杂度: O(n) 。

空间复杂度: O(n) 。

一次哈希表:

上面我们是先把数组放到哈希表中,然后再进行遍历,实际上我们可以一边放一边进行遍历操作,直到某一个时刻,打成我们的目标,这时我们可以直接返回数据,剩下没有放到哈希表中的数据也不用再往里面放了。

public int[] twoSum_3(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement)) {
            return new int[] { map.get(complement), i };
        }
        map.put(nums[i], i);
    }
    throw new IllegalArgumentException("No two sum solution");
}

虽然这种方式看着比前面的两次哈希表更加的高效,实际上时间复杂度和空间复杂度是一致的,同样是:

时间复杂度: O(n) 。

空间复杂度: O(n) 。

posted @ 2020-07-29 08:50  极客挖掘机  阅读(460)  评论(0编辑  收藏  举报