leectcode15:三数之和

题目描述

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。

注意:

答案中不可以包含重复的三元组。
数组中的数字可以是小数或负数。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],

一个可能的解集为:

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

解题思路

首先对数组进行排序,这样可以方便处理重复元素和减少搜索范围。
之后使用三个指针,i,left,right。其中 i 从数组的起始位置开始遍历,left 初始化为 i+1,right 初始化为数组末尾。

  • 对于每个 i,固定它,然后移动 left 和 right 来查找满足 nums[i] + nums[left] + nums[right] == 0 的三元组。
  • 如果和大于 0,则 right-- 以减小和;如果和小于 0,则 left++ 以增大和;如果和等于 0,则找到了一个解,并移动 left 和 right 来避免重复解,同时将它们添加到结果中。
  • 在移动 left 和 right 时,需要跳过重复的元素以避免重复解。
    跳过重复元素:
  • 在移动 i 时,如果 nums[i] 等于 nums[i-1],则跳过该元素,以避免重复解。
  • 在移动 left 时,如果 nums[left] 等于 nums[left-1],则跳过该元素。
  • 在移动 right 时,如果 nums[right] 等于 nums[right+1],则跳过该元素。

解题示例 (JavaScript)

var threeSum = function(nums) {  
    const result = [];  
    nums.sort((a, b) => a - b); // 排序  
  
    for (let i = 0; i < nums.length - 2; i++) {  
        // 跳过重复元素  
        if (i > 0 && nums[i] === nums[i - 1]) continue;  
  
        let left = i + 1;  
        let right = nums.length - 1;  
  
        while (left < right) {  
            const sum = nums[i] + nums[left] + nums[right];  
            if (sum === 0) {  
                result.push([nums[i], nums[left], nums[right]]);  
                  
                // 跳过重复元素  
                while (left < right && nums[left] === nums[left + 1]) left++;  
                while (left < right && nums[right] === nums[right - 1]) right--;  
                  
                left++;  
                right--;  
            } else if (sum < 0) {  
                left++;  
            } else {  
                right--;  
            }  
        }  
    }  
  
    return result;  
};

你可能感兴趣的:(Leetcode题解,算法,数据结构,排序算法)