JS解决查找表与求和问题

两数之和

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

示例：

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

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

• 方法一：暴力解法

遍历所有数据对，判断是否等于 target，时间复杂度度 O(n^2)

//执行用时 :124 ms, 在所有 javascript 提交中击败了54.32%的用户
//内存消耗 :35.2 MB, 在所有 javascript 提交中击败了21.37%的用户
var twoSum = function (nums, target) {
  var res = [];
  for (let i = 0; i < nums.length; i++) {
    for (let j = i + 1; j < nums.length; j++) {
      if (nums[i] + nums[j] == target) {
        res.push(i, j);
        return res;
      }
    }
  }
};

• 方法二：查找表

//dict
//执行用时 :52 ms, 在所有 javascript 提交中击败了99.77%的用户
//内存消耗 :34.9 MB, 在所有 javascript 提交中击败了30.64%的用户
var twoSum = function (nums, target) {
  let res = {};
  for (let i = 0; i < nums.length; i++) {
    if (res.hasOwnProperty(nums[i])) {
      return [res[nums[i]], i];  //找到i下标对应要找的值，则返回
    } else {
      res[target - nums[i]] = i; //保存对应i下标要找的值
    }
  }
};

//数组
//执行用时 :152 ms, 在所有 javascript 提交中击败了33.55%的用户
//内存消耗 :34.3 MB, 在所有 javascript 提交中击败了84.79%的用户
var twoSum = function (nums, target) {
  for (let i = 0; i < nums.length; i++) {
    let result = nums.lastIndexOf(target - nums[i]);
    if (result !== -1 && result !== i) {
      return [i, result]
    }
  }
};

三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。注意：答案中不可以包含重复的三元组。

示例：

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

• 双指针

//执行用时 :196 ms, 在所有 javascript 提交中击败了68.70%的用户
//内存消耗 :46.8 MB, 在所有 javascript 提交中击败了57.55%的用户
var threeSum = function (nums) {
  if (!nums || nums.length < 3) return [];
  nums.sort((a, b) => { return a - b; });

  let res = [];
  for (let i = 0; i < nums.length - 2; i++) {
    let l = i + 1;
    let r = nums.length - 1;

    //去重
    if (i > 0 && nums[i] === nums[i - 1]) continue;

    while (l < r) {
      let sum = nums[i] + nums[l] + nums[r];
      if (sum < 0) {
        l++;
      } else if (sum > 0) {
        r--;
      } else {
        res.push([nums[i], nums[l], nums[r]]);
        while (nums[l] === nums[l + 1]) l++;
        while (nums[r] === nums[r - 1]) r--;
        l++;
        r--;
      }
    }
  }
  return res;
};

• 暴力解法超时：311 / 313 个通过测试用例

var threeSum = function (nums) {
  if (nums.length < 3) {
    return [];
  }
  let res = [];
  let key = {};
  for (let i = 0; i < nums.length; i++) {
    for (let j = i + 1; j < nums.length; j++) {
      let index = nums.indexOf(-(nums[i] + nums[j]), j + 1);
      if (index != -1) {
        let temp = [nums[i], nums[j], nums[index]].sort((a, b) => { return a - b })
        let str = temp.join('');
        if (!key.hasOwnProperty(str)) {
          key[str] = 1;
          res.push(temp);
        }
      }
    }
  }
  return res;
};

四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。注意：答案中不可以包含重复的四元组

示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

• 在三数之和的基础上添加一层循环

//执行用时 :96 ms, 在所有 javascript 提交中击败了95.30%的用户
//内存消耗 :37.1 MB, 在所有 javascript 提交中击败了42.64%的用户
var fourSum = function (nums, target) {
  if (!nums || nums.length < 4) return [];
  nums.sort((a, b) => { return a - b; });

  let res = [];
  for (let i = 0; i < nums.length; i++) {
    let threeSum = target - nums[i];
    if (i > 0 && nums[i] == nums[i - 1]) { continue; }

    for (let j = i + 1; j < nums.length - 2; j++) {
      let l = j + 1;
      let r = nums.length - 1;

      if (j > i + 1 && nums[j] == nums[j - 1]) { continue; }
      while (l < r) {
        let sum = nums[j] + nums[l] + nums[r];
        if (sum < threeSum) {
          l++;
        } else if (sum > threeSum) {
          r--;
        } else {
          res.push([nums[i], nums[j], nums[l], nums[r]]);
          while (nums[l] === nums[l + 1]) l++;
          while (nums[r] === nums[r - 1]) r--;
          l++;
          r--;
        }
      }
    }
  }
  return res;
};

• 双指针+Set去重

//执行用时 :124 ms, 在所有 javascript 提交中击败了48.90%的用户
//内存消耗 :38 MB, 在所有 javascript 提交中击败了24.81%的用户

var fourSum = function (nums, target) {
  let set = new Set()
  nums.sort((a, b) => a - b)
  for (let i = 0; i < nums.length - 3; i++) {
    for (let j = i + 1; j < nums.length - 2; j++) {
      let start = j + 1, end = nums.length - 1
      while (start < end) {
        let sum = nums[i] + nums[j] + nums[start] + nums[end]
        if (sum < target) {
          start++
        } else if (sum > target) {
          end--
        } else {
          set.add([nums[i], nums[j], nums[start], nums[end]].join(','))
          start++
          end--
        }
      }
    }
  }
  return [...set].map(v => v.split(',').map(v => +v))
};

• 固定中间两个数然后不断逼近

//执行用时 :124 ms, 在所有 javascript 提交中击败了48.90%的用户
//内存消耗 :37.7 MB, 在所有 javascript 提交中击败了31.01%的用户
var fourSum = function (nums, target) {
  if (nums.length < 4)
    return [];
  nums.sort((a, b) => a - b);
  var hash = new Map();
  for (var midF = 1; midF < nums.length - 2; midF++) {
    for (var mid = midF + 1; mid < nums.length - 1; mid++) {
      if (mid - 1 !== midF && nums[mid] === nums[mid - 1])
        continue;
      var start = 0,
        end = nums.length - 1;
      while (start < midF && end > mid) {
        if (nums[start] + nums[midF] + nums[mid] + nums[end] === target) {
          var arr = [nums[start], nums[midF], nums[mid], nums[end]],
            str = arr.join('-');
          if (hash.has(str) === false) {
            hash.set(str, arr);
          }
        }
        if (nums[start] + nums[midF] + nums[mid] + nums[end] > target)
          end -= 1;
        else
          start += 1;
      }
    }
  }
  return [...hash].map((a) => a[1]);
};

来源：力扣（LeetCode）https://leetcode-cn.com