feat: [18] 四数之和

src/leecode/javascript/18.四数之和.js

@@ -0,0 +1,59 @@
+/*
+ * @lc app=leetcode.cn id=18 lang=javascript
+ *
+ * [18] 四数之和
+ */
+
+// @lc code=start
+/**
+ * @param {number[]} nums
+ * @param {number} target
+ * @return {number[][]}
+ */
+
+var fourSum = function(nums, target) {
+  nums.sort((a, b) => a - b);
+  // n 为 4，从 nums[0] 开始计算和为 target 的四元组
+  return nSumTarget(nums, 4, 0, target);
+};
+
+/* 注意：调用这个函数之前一定要先给 nums 排序 */
+// n 填写想求的是几数之和，start 从哪个索引开始计算（一般填 0），target 填想凑出的目标和
+var nSumTarget = function(nums, n, start, target) {
+  var sz = nums.length;
+  var res = [];
+  // 至少是 2Sum，且数组大小不应该小于 n
+  if (n < 2 || sz < n) return res;
+  // 2Sum 是 base case
+  if (n === 2) {
+    // 双指针那一套操作
+    var lo = start, hi = sz - 1;
+    while (lo < hi) {
+      var sum = nums[lo] + nums[hi];
+      var left = nums[lo], right = nums[hi];
+      if (sum < target) {
+        while (lo < hi && nums[lo] === left) lo++;
+      } else if (sum > target) {
+        while (lo < hi && nums[hi] === right) hi--;
+      } else {
+        res.push([left, right]);
+        while (lo < hi && nums[lo] === left) lo++;
+        while (lo < hi && nums[hi] === right) hi--;
+      }
+    }
+  } else {
+    // n > 2 时，递归计算 (n-1)Sum 的结果
+    for (var i = start; i < sz; i++) {
+      var sub = nSumTarget(nums, n - 1, i + 1, target - nums[i]);
+      for (var arr of sub) {
+        // (n-1)Sum 加上 nums[i] 就是 nSum
+        arr.push(nums[i]);
+        res.push(arr);
+      }
+      //跳过重复元素
+      while (i < sz - 1 && nums[i] === nums[i + 1]) i++;
+    }
+  }
+  return res;
+};
+// @lc code=end