18.leetCode 四数之和——极致优化

执行用时分布
87ms
击败85.10%使用 TypeScript 的用户

消耗内存分布
54.75MB
击败79.81%使用 TypeScript 的用户

function fourSum(nums: number[], target: number): number[][] {
  //排序
  nums.sort((a, b) => a - b);
  const size = nums.length - 1;

  //长度小于4，直接返回
  if (nums.length < 4) {
    return [];
  }

  //最小的四个值相加大于 target,则直接返回
  if (nums[0] + nums[1] + nums[2] + nums[3] > target) {
    return [];
  }
  //最大的四个值相加小于 target,则直接返回
  if (nums[size - 3] + nums[size - 2] + nums[size - 1] + nums[size] < target) {
    return [];
  }

  const ans = [];

  //  第一层循环
  for (let i = 0; i < size - 2; i++) {
    // 除首次外，如果内容重复则跳过
    if (i > 0 && nums[i] === nums[i - 1]) {
      continue;
    }
    //最小的四个值相加大于 target,则退出循环
    if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) {
      break;
    }
    //当前索引和最大的三个值相加大于 target,则跳过本次循环
    if (nums[i] + nums[size - 2] + nums[size - 1] + nums[size] < target) {
      continue;
    }

    for (let j = i + 1; j < size - 1; j++) {
      // 除首次外，如果内容重复则跳过
      if (j > i + 1 && nums[j] === nums[j - 1]) {
        continue;
      }

      //最小的四个值相加大于 target,则退出循环
      if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target) {
        break;
      }
      //当前索引和最大的三个值相加大于 target,则跳过本次循环
      if (nums[i] + nums[j] + nums[size - 1] + nums[size] < target) {
        continue;
      }

      //   双指针
      let left = j + 1;
      let right = size;
      while (left < right) {
        const sum = nums[i] + nums[j] + nums[left] + nums[right];
        if (sum === target) {
          ans.push([nums[i], nums[j], nums[left], nums[right]]);
        }

        if (sum >= target) {
          while (nums[right] === nums[--right]) {}
        } else {
          while (nums[left] === nums[++left]) {}
        }
      }
    }
  }
  return ans;
}

console.log(fourSum([2, 2, 2, 2, 2], 8));

//[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]