一、题目来源:leetCode 15. 三数之和
题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
解题思路
- 第一种是暴力解法,三重循环,但是很容易超时,特别是当所有数都相等的最坏情况下,时间复杂度为 O(n^3)
- 第二种是采用双指针解法,先把所有数进行排序,循环遍历数,以一个数为基准,去寻找两外两个数,使得三个数相加和为0.
本题解决的难度在于去掉重复的计算,因为数进行了排序,所有当遍历到当前数 > 0的时候直接返回之前的结果就可以了,后面的数怎么找两个数都不会 小于 0 了。
如果三个数之和小于0,那么左指针向右挪,如果三数之和大于0,那么右指针向左挪。
具体的代码实现:
var threeSum = function(nums) {
nums.sort((a,b)=>{
return a-b
})
let len = nums.length
if(len < 3) {
return []
}
let arr = []
for(let i = 0; i < len; i ++) {
if(nums[i] > 0) return arr // 因为排序了,所有后面的正数之和不会等于0了
if(i > 0 && nums[i] == nums[i - 1]) continue // 去重
let l = i + 1;
let r = len - 1
while (l < r) {
let sum = nums[l] + nums[r] + nums[i]
if(sum == 0) {
arr.push([nums[l],nums[r], nums[i]])
while( l < r && nums[l] == nums[l+1]) l++; // 去重
while(l < r && nums[r] == nums[r-1]) r--; // 去重
l++;
r--;
}
else if(sum < 0) l++
else if(sum > 0) r--
}
}
return arr
};