leetcode 第15题 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
// 示例
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
// 双指针 + 排序
// 主要难点在于考虑什么时候该去重
// 时间复杂度 O(n^2)
function threeSum(nums) {
const lenght = nums.length; // 数组长度
let result = [] // 返回结果的容器
nums.sort((a, b) => a - b) // 双指针 + 排序的 排序 这一步
if (nums === null || lenght < 3) { // 特殊情况的处理
return []
}
for (let i = 0; i < lenght; i++) {
// 需要去重 两个数相同就没必要重复之前的循环了
if (i > 0 && nums[i] === nums[i - 1]) {
continue
}
// i都大于0了,因为我们的数组是排过序的,
// i后面的数字就肯定都大于0是吧,可以直接返回了
if (nums[i] > 0) {
return result
}
let l = i + 1, // 左指针 从i下一位开始
r = lenght - 1; // 右指针 从最后一位开始;
while (l < r) { // 循环条件
const sum = nums[i] + nums[l] + nums[r];
if (sum === 0) {
result.push([nums[i], nums[l], nums[r]])
while (l < r && nums[l] === nums[l + 1]) {
l++
}
l++
while (l < r && nums[r] === nums[r - 1]) {
r--
}
r--
} else if (sum < 0) {
// 加起来的数小于target 说明最小的数太小了
l++
} else {
// 加起来的数大于target, 说明最大的数太大了
r--
}
}
}
return result
}