算法实践 - 三数相加
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
这道题最简单的方法是3个for循环,对数组进行遍历,比较每种情况,最后输出满足条件的结果。使用这种方法的弊端可想而知,时间复杂度非常高(n^n)
比较理想的做法是对数组进行排序后使用前后指针查找,以此减少时间复杂度
解题步骤:(解题不唯一)
1、遍历数组
a) 根据元素大小判断该情况是否满足要求
b) 元素查重
2、初始化前后指针,while条件循环,
a) 判断该情况是否存在
b) 当情况满足,判定重复元素的情况,满足则加入结果数组
c) 小于 || 大于 情况分别加减前后指针索引位置
3、当左右指针汇合时 返回结果数组
function threeNum(arr) {
if(!arr || arr.length < 3) return [];
arr.sort((a, b) => a - b); // 排序
let result = [];
for (let i = 0; i < arr.length; i++) {
if(arr[i] === 0 && arr[i+1] !== arr[i]) break; // 如果当前数字等于0,则前数大于0,结束循环
if(arr[i] > 0) break; // 首索引 > 0 直接退出,不可能3数 num 为0
if(i > 0 && arr[i] == arr[i-1]) continue; // 判断重复的元素
let left = i+1;
let right = arr.length-1;
while(left < right){
const arrRes = arr[i] + arr[left] + arr[right];
if(arrRes == 0){
result.push([arr[i],arr[left],arr[right]]);
while (arr[left] == arr[left+1]) left++; // 判重
while (arr[right] == arr[right-1]) right--; // 判重
left++;
right--;
}
else if (arrRes < 0) left++;
else if (arrRes > 0) right--;
}
}
return result;
}
大功告成