力扣第一题:两数之和

需求

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

提示：
2 <= nums.length <= 104
-109 <= nums[i] <= 109
-109 <= target <= 109
只会存在一个有效答案

进阶：你可以想出一个时间复杂度小于 O(n2) 的算法吗？

分析

我们可以一边遍历数组, 一边将每个值与 target 做运算. 比如 target 是7, 数组中的第一个元素是 3 , 那么就可以将 target - 3 的值存进到 Map 里去, key 是 target-3, value是当前数组的下标.
这样我们就可以在遍历的时候一遍存,一边取出来作比较, 满足返回下标即可.

代码

public int[] twoSum(int[] nums, int target) {
    int length = nums.length;
    if( length == 2 ){
        return new int[]{0, 1};
    }
    Map<Integer, Integer> map = new HashMap(nums.length);
    for( int i = 0; i < length; i++ ){
        int num = nums[i];
        //  先判断
        if( map.containsKey(num) ){
            return new int[]{i, map.get(num)};
        }
        //  存值
        map.put(target - num, i);
    }
    return null;
}

代码解释
遍历 nums 数组:

for (int i = 0; i < length; i++) {
    int num = nums[i];
    // 先判断
    if (map.containsKey(num)) {
        return new int[]{i, map.get(num)};
    }
    // 存值
    map.put(target - num, i);
}

对于 nums 数组中的每一个元素 num，检查 map 中是否存在 num 作为键。如果存在，说明之前的某个数字与当前的 num 的和等于 target。此时返回当前索引 i 和 map 中存储的索引 map.get(num)。
如果 num 不在 map 中，将 target - num 和当前索引 i 存储到 map 中。这样，map 将记录下当前数字需要的另一个数 target - num 的位置。

没有找到结果时返回 null:
return null;
如果遍历完所有元素都没有找到符合条件的两个数，则返回 null。

执行结果:

结尾

以上 是我对这道算法的一些遐想和延伸, 可能不是最优解, 但是算法的优化嘛 本身就是一个思索的过程, 能在这个思索和迭代的过程中有所收获和乐趣就是在成长了, 欢迎大家一起来交流更多的解答…