该题目是来自力扣(LeetCode)上的一道算法题。
题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]
方法一:暴力破解
首先拿到题目直观能想到的肯定是暴力破解,虽然感觉可能会超时,不过还是想先试一下(笨方法也是方法,笔者的水平还达不到一上来就能想到最优解的方法),说干就干。
var threeSum = function(nums) {
let arr = nums.sort((a, b) => a-b); // 先排序,方便后面代码过滤掉重复项
let arrays = [];
const len = nums.length;
for(let i = 0; i < len; i++) {
// 从第二位开始,如果当前项等于上一项,则跳过
if (i!== 0 && arr[i] === arr[i-1]) {
continue;
}
for(let j = i+1; j < len; j++) {
if (j!== i+1 && arr[j] === arr[j-1]) { // 跳过
continue;
}
for(let k = j+1; k < len; k++) {
if (k!==j+1 && arr[k] === arr[k-1]) { // 跳过
continue;
}
if (arr[i]+arr[j]+arr[k] === 0) {
arrays.push([arr[i],arr[j],arr[k]])
}
}
}
}
return arrays;
};
这个方法虽然可行,但是,对于数据量太大的测试用例则会超时。推荐采取下一种方法。
方法二:双指针
将第二层循环和第三层循环一起来做,定义两个循环变量,一个从左往右,一个从右往左,当满足一定条件时进行左右移动,这就是传说中的双指针。
var threeSum = function(nums) {
let arr = nums.sort((a, b) => a-b); // 从小到大排序
let arrays = [];
const len = nums.length;
for(let i = 0; i < len; i++) {
// 因为是按从小到大排序的,如果当前项都比0大了,那之后的数更不用说了,直接返回当前结果数组
if (arr[i] > 0) { return arrays; }
// 从第二位开始,如果当前项等于上一项,则跳过,避免重复
if (i > 0 && arr[i] === arr[i-1]) { continue; }
// 定义左指针
let l = i+1;
// 定义右指针
let r = len-1;
while(l < r) {
const count = arr[i]+arr[l]+arr[r];
if (count > 0) { // 如果当前和大于0,则右指针向前移动
r--;
} else if (count < 0) { // 如果当前和小于0,则左指针向后移动
l++;
} else {
// 满足条件时,存到结果集
arrays.push([arr[i],arr[l],arr[r]]);
// 提前判断一下如果下一项和当前项相同,则继续往后移动
while(l < r && arr[l] === arr[l+1]){
l++;
}
// 提前判断一下如果前一项和当前项相同,则继续往前移动
while(l < r && arr[r] === arr[r-1]){
r--;
}
l++;
r--;
}
}
}
return arrays;
};
双指针解法是看官网题解以及网友评论才做出来的,这里学习记录一下,加深印象,如有不对或更好的方法,欢迎指正。