Leetcode18四数之和(双指针法)

·  阅读 58

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

0 <= a, b, c, d < n a、b、c 和 d 互不相同 nums[a] + nums[b] + nums[c] + nums[d] == target 你可以按 任意顺序 返回答案 。 答题:

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function (nums, target) {
  let res = []
  if (nums.length < 4) {
    return res
  }
  let len = nums.length;
  nums = nums.sort((a, b) => {
    return a - b
  })
  for (let i = 0; i < len - 1; i++) {

    if (i > 0 && nums[i] === nums[i - 1]) {
      continue
    }
    if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target) {
      return res
    }
    for (let j = len - 1; j > i + 2; j--) {
      if (nums[j] === nums[j + 1]) {
        continue;
      }

      let left = i + 1
      let right = j - 1
      while (left < right) {
        //
        let sum = nums[i] + nums[left] + nums[right] + nums[j]
        if (sum === target) {
          res.push([nums[i], nums[left], nums[right], nums[j]])
          while (left < right && nums[left] === nums[left + 1]) {
            left++
          }
          while (left < right && nums[right] === nums[right - 1]) {
            right--
          }
          left++
          right--
        } else if (sum < target) {
          left++
        } else {
          right--
        }
      }
    }
  }
  return res

};
复制代码

四数之和跟三数之和挺相似的,都要有一个排序的过程,但是四数之和多了一个右侧边界的收缩。 同时中间两个数字不断地尝试,找到target即可添加到res中。 向右移动左边界的时候,首先计算一下左侧四个数字的和,如果和大于目标选项,直接输出,不用再继续后面的for循环了。 同时由于数字选取不能相同,假设有一个结果第一个数选择了1那么for循环中如果nums[i]就不应该再有1出现了。 举个例子 [2,2,1,2,2]我们选择满足target=5的四个数 在边界左移的过程中会出现 [2,1,2,2]两次但这个明显不符合要求,重复了毕竟 所以函数中要包括选项的去重 于是在第一层for循环第二次for循环以及while循环中都有去重操作 具体打印一下代码,体验整个过程!

分类:
前端
标签:
收藏成功!
已添加到「」, 点击更改