中等算法给你一个包含 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]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
作者:力扣 (LeetCode)
链接:https://leetcode-cn.com/leetbook/read/top-interview-questions-medium/xvpj16/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function (nums) {
if (nums.length < 3) {
return [];
}
const arr = nums.sort((prev, curt) => {
return prev - curt;
});
const ans = new Array();
const l = arr.length;
for (let i = 0; i < l - 2; i++) {
if (i > 0 && arr[i] === arr[i - 1]) {
continue;
}
if (arr[i] > 0) {
break;
}
let left = i + 1;
let right = l - 1;
const target = 0 - arr[i];
while (left < right) {
const t = arr[left] + arr[right];
if (t === target) {
ans.push([arr[i], arr[left], arr[right]]);
while (left < right && arr[left] === arr[right]) {
left++;
}
while (left < right && arr[right] === arr[right - 1]) {
right--;
}
left++;
right--;
} else if (t < target) {
left++;
} else {
right--;
}
}
}
return ans;
};