问题
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路
- 首先给的数组要有三个数,不到三个数就返回空数组。且三个数之和为 0 则三个数中必有负数与正数,可能有 0。
- 对数组进行升序排序,遍历数组,每遍历一个元素,就定义左右两个指针,左指针是当前元素的下一个元素,右指针是尾部元素。
- 若当前元素是大于 0 的,那当前元素与左右指针指向的元素之和必定是大于 0,跳出循环,当前元素与上一个元素值相同则跳过这一轮继续下一轮遍历。
- 循环当左指针小于右指针时,求当前元素与左右指针指向的元素之和,和为 0 ,就把三个元素组成一个数组存起来,左右指针移动过程中有相同值时只记录一个,这样就可以去重了;和小于 0,则左指针指向元素过小,左指针右移;和大于 0,则右指针指向的元素值过大,右指针左移。当然左指针要小于右指针。
代码实现
const threeSum = nums => {
let len = nums.length
let threeNumSum = []
if (len < 3) return threeNumSum
nums.sort((a, b) => a - b)
// 遍历升序后的数组,每遍历到一个元素,固定这个元素,定义双指针,计算三数之和
for (let i = 0;i < len; i++) {
// 三数之和等于 0 必有负数
if (nums[i] > 0) break
// 若此值与上一个值相同则继续下一轮循环(只有 i > 0 时有上一个值)
if (i > 0 && nums[i] === nums[i - 1]) continue
let L = i + 1, R = len - 1
while (L < R) {
const sum = nums[i] + nums[L] + nums[R]
if (sum === 0) {
threeNumSum.push([nums[i], nums[L], nums[R]])
while (L < R && nums[L] === nums[L + 1]) L++ // 去重
while (L < R && nums[R] === nums[R - 1]) R-- // 去重
// 继续寻找
L++
R--
}
// 若三数之和大于 0,则 right 左移;若小于 0 ,则 left 右移;三数之和等于 0,放入答案数组
else if (sum < 0) L++
else if (sum > 0) R--
}
}
return threeNumSum
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum