力扣题目链接[1]
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
「示例 1:」
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
「提示:」
0 <= nums.length <= 3000
-10^5 <= nums[i] <= 10^5
思路:
本题是两数之和的升级版,但是思路和两数之和不尽相同。
本题采用排序+双指针的方式题解。首先确保数组是有序数组,然后固定一个元素,使用双指针分别指向固定元素的下一个元素和数组末尾元素,将三者元素相加进行判断。因为数组是有序的,所以和大于0时,左移右指针;和小于0时,右移左指针。等于0时,将结果放入结果数组。
由于题目要求不能包含重复的集合,因此需要去重处理。当遇到前后相同元素时,要直接跳过。
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
let length = nums.length; // 缓存数组长度
if (!nums || length < 3) return []; // 提前返回
let result = []; // 初始化结果数组
let left = null; // 初始化左指针
let right = null; // 初始化右指针
nums.sort((a, b) => a - b); // 对数组进行排序
for (let i = 0; i < length; i++) {
if (nums[i] > 0) break; // 如果第一个数就大于0,那么后续相加肯定大于0,提前返回
if (i > 0 && nums[i] === nums[i - 1]) continue; // 去重
left = i + 1; // 左指针指向当前索引的下一个索引
right = length - 1; // 右指针指向数组最后一个元素
while(left < right) { // 循环终止条件是左指针大于等于右指针
let num = nums[i] + nums[left] + nums[right]; // 三数之和
if (!num) { // 和等于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 (num < 0) left++; // 和大于0,右移左指针
else if (num > 0) right--; // 和小于0,左移右指针
}
}
return result;
};
[1]
力扣题目链接: https://leetcode-cn.com/problems/3sum/