【算法之路】Leecode No.1 两数之和

出处

Leecode No.1

描述

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例

给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

分析

暴力解法

这道题不难，大多数人都可以直接想到最简单的暴力解法。
依次取出数组中的每个元素，然后对数组遍历看是否有能符合的加数即可。

HashMap解法

利用HashMap存储已经遍历的数和下标，然后只需要遍历一次，遍历到某元素n时，查找Map中是否有(target - n)，如果有，说明有结果，直接返回n的下标，和map中对应元素的下标即可。

过程追踪

暴力解法很容易理解，这里就不详细展开了。我们直接看利用HashMap的解法，以示例为例，

定义用于存储已经遍历过的HashMap
已知条件：nums = [2, 7, 11, 15], target = 9
遍历数组，

• 当遍历到2时，从map中查找是否含有（9 - 2 =）7, map中没有，直接把当前遍历的元素存到map中，key为元素值（2），value为数组下标（0），继续遍历
• 当遍历到7时，从map中查找是否还有（9 - 7 =）2，map中含有，说明找到了答案，直接返回当前元素（7）的下标（1），和找到元素（2）的下标（2），算法完。返回[0, 1]

题解

暴力法

for (int i = 0;i < nums.length;i++) {
    int x = nums[i];
    for (int j = i + 1; j < nums.length;j++) {
        if (x + nums[j] == target) {
            return new int[] {i, j};
        }
    }
}
return new int[] {0, 0};

代码消耗

利用HashMap方法

HashMap<Integer, Integer> adds = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
    int current = nums[i];
    int res = target - current;
    if (adds.containsKey(res)) {
        return new int[] {i, adds.get(res)};
    }
    adds.put(current, i);
}
return new int[] {0, 0};

代码消耗