题目:
给你一个包含 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]]
思路:(注意:不重复的三元数组,所以需要不断判断来去重的)
首先,将数组从小到大sort一下排序,令i从第一位开始固定,从i往后设置两个指针(头指针与尾指针),使这两个指针的和相加恰好为i所对应的相反数。注意:如果i为正数的话直接跳出循环,因为正数相加什么都不可能为0),操作i指针时对i进行查重处理,即如果第i与第i-1个对应的数字相等,则continue。left与right指针开始操作,当三者相加为0时,将三个数拍成数组的形式,添加到一个空数组中(最后结果的数组)。由于碰到了相加恰好为0的数字们,于是需要继续向下操作,即查重,再次遇到相同的就需要left++,与right--;查重之后,再指向下一个数字重新开始相加比较是否为0,往返操作。 当三者相加大于0时,则左指针向前移动一位(以减小sum的总值)。当三者相加小于0时,则右指针向后移动一位(以增加sum的总值)。最后返回我们的结果数组。
以上三个参数都要进行去重处理,i是在初始操作时去重,left与right是在确定了三者和相加为0时去重 然后再分别指向下一个数字。
代码实现:
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function (nums) {
var arr = []
nums.sort((a, b) => a - b)
for (let i = 0; i < nums.length; i++) {
if (nums[i] > 0) break;
if (i > 0 && nums[i] == nums[i - 1]) continue;
var left = i + 1, right = nums.length - 1
while (left < right) {
const sum = nums[i] + nums[left] + nums[right]
if (sum == 0) {
arr.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 if (sum > 0) right--;
}
}
return arr;
};
欢迎兄弟姐妹一起来讨论咯