【leetcode刷题笔记】01.两数之和

题目详情：链接地址：https://leetcode-cn.com/problems/two-sum

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

分析与解答：

第一反应是遍历数组，每次取nums[i]后，再去遍历数组，查找index不等于i且值为target-nums[i]的下标。这样导致的后果是算法复杂度为O(n^2),因为遍历了两遍数组。初次代码如下：

var twoSum = function(nums, target) {
    const result = [];
    for (let i = 0; i < nums.length; i++) {
        const firstIndex = nums.findIndex((value, index) => {
            return index != i && value === target - nums[i];
        });
        if (firstIndex !== -1) {
            result[0] = i;
            result[1] = firstIndex;
            break;
        }
    }
    return result;
};

 我们可以使用hash表来解决这个问题，只需要遍历一次数组即可。新建一个Map对象，当Map对象中有target-nums[i]这个key时，则返回[i, map.get(target-nums[i])]，否则设置该Map对象key为当前数组值nums[i]， value为当前数组下标i。代码如下：

var twoSum = function(nums, target) {
    // 想法： map中key为值，value为数组中的下表
    var newMap = new Map();
    for(var i = 0; i < nums.length; i++) {
        if (newMap.has(target - nums[i])) {
            return [newMap.get(target - nums[i]), i];
        } else {
            newMap.set(nums[i], i)
        }
    }
    return [];
}

性能明显提升了数倍

写在最后，因为TS逐渐成为前端开发必备技能，所以TS版本如下：

function twoSum(nums: number[], target: number): number[] {
    let newMap = new Map();
    for (let i = 0; i < nums.length; i++) {
        if (newMap.has(target-nums[i])) {
            return [newMap.get(target-nums[i]), i];
        } else {
            newMap.set(nums[i], i);
        }
    }
    return []
};