100道面试必会算法-14-三数之和
题目描述
一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
**注意:**答案中不可以包含重复的三元组。
解题思路
还有一种思想是:将其中一个当做目标值放入hashmap中进行查找,其他两个用两层for循环进行嵌套遍历,不过有问题就是有很多无效解需要判断,因此采用下方的思路,传送门:https://leetcode.cn/problems/3sum/solutions/11525/3sumpai-xu-shuang-zhi-zhen-yi-dong-by-jyd/
-
先将
nums
排序,时间复杂度为O(NlogN)
。 -
固定
3
个指针中最左(最小)元素的指针 k,双指针 i,j 分设在数组索引(k,len(nums))
两端。 -
双指针
i
,j
交替向中间移动,记录对于每个固定指针 k 的所有满足nums[k] + nums[i] + nums[j] == 0
的i,j
组合: -
当
nums[k] > 0
时直接break
跳出:因为nums[j] >= nums[i] >= nums[k] > 0
,即3
个元素都大于0
,在此固定指针k
之后不可能再找到结果了。 -
当
k > 0
且nums[k] == nums[k - 1]
时即跳过此元素nums[k]
:因为已经将nums[k - 1]
的所有组合加入到结果中,本次双指针搜索只会得到重复组合。 -
i,j
分设在数组索引(k,len(nums))
两端,当i < j时循环计算s = nums[k] + nums[i] + nums[j]
,并按照以下规则执行双指针移动:
当s < 0
时,i += 1
并跳过所有重复的nums[i]
;
当s > 0
时,j -= 1
并跳过所有重复的nums[j]
;
当s == 0
时,记录组合[k, i, j]
至res
,执行i += 1
和j -= 1
并跳过所有重复的nums[i]
和nums[j]
,防止记录到重复组合。
代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
for(int k = 0; k < nums.length - 2; k++){
if(nums[k] > 0) break;
if(k > 0 && nums[k] == nums[k - 1]) continue;
int i = k + 1, j = nums.length - 1;
while(i < j){
int sum = nums[k] + nums[i] + nums[j];
if(sum < 0){
while(i < j && nums[i] == nums[++i]);
} else if (sum > 0) {
while(i < j && nums[j] == nums[--j]);
} else {
res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
while(i < j && nums[i] == nums[++i]);
while(i < j && nums[j] == nums[--j]);
}
}
}
return res;
}
}