三数之和
题目:
判断一个整数数组nums,判断是否存在三元组[nums[i],nums[j],nums[k]],满足i!=j, i!=k,j!=k, 同时满足nums[i]+nums[j]+nums[k] = 0。 返回所有和为0且不重复的三元组
示例:
示例1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
思路:
1、三数相加等于0 必须得有小于零的数 所以先排序 如果第一个数大于零 直接返回
2、设置数组第一位为i,后面两个指针的和等于前面i的相反数
3、对i进行查重,如果第i与第i-1个数字相等,continue
4、另外两个指针操作,三者相加等于0,将三个数放到一个数组里
5、由于碰到了相加恰好为0的数字们,于是需要继续向下操作,即查重,再次遇到相同的就需要left++,与right–;查重之后,再指向下一个数字重新开始相加比较是否为0,往返操作。
6、 三者相加大于0时,则左指针向前移动一位(以减小sum的总值)。当三者相加小于0时,则右指针向后移动一位(以增加sum的总值)。最后返回我们的结果数组。
代码:
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;