数据结构和算法 -- 两数之和

112 阅读1分钟

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

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

示例:

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

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

方法1:双for循环

const twoSum = function(num, target) {
    for (let i = 0; i < num.length; i++) {
        const diff = target - num[i];
        for (let j = i + 1; j < num.length; j++) {
            if (num[j] === diff) {
                return [i, j];
            }
        }
    }
}

console.log(twoSum([3, 8, 11, 5, 13, 6], 18)); // [3, 4]

方法2:使用Map函数

const twoSum = function(num, target) {
    const targetMap = new Map();
    for (let i = 0; i < num.length; i++) {
        const key = target - num[i];
        if (targetMap.has(key)) {
            return [targetMap.get(key), i];
        }
        targetMap.set(num[i], i);
    }
}

console.log(twoSum([3, 8, 11, 5, 13, 6], 18)); // [3, 4]