15. 三数之和
难度中等2596
给你一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
通过次数330,688
提交次数1,119,900
题解
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
for(int i=0; i < nums.length; i++){
int target = 0 - nums[i];
int left = i+1;
int right = nums.length-1;
if(nums[i] >0) break; //如果当前的值都>0,肯定不会再和其他的2个数相加为0
if(i==0 || nums[i] != nums[i-1]){ //相同数字去重
while( left < right){
if( nums[left] + nums[right] == target){
List<Integer> temp = new ArrayList<>();
temp.add(nums[i]);
temp.add(nums[left]);
temp.add(nums[right]);
res.add(temp);
//相同数字去重
while(left< right && nums[left] == nums[left+1]) left++;
while(left< right && nums[right] == nums[right-1]) right--;
left++;
right--;
}else if( nums[left] + nums[right] > target){
right--;
}else{
left++;
}
}
}
}
return res;
}
}
注意去重, 相同数字直接略过
双指针法思路: 固定 3个指针中最左(最小)数字的指针 k,双指针 i,j 分设在数组索引 (k, len(nums))(k,len(nums)) 两端,通过双指针交替向中间移动,记录对于每个固定指针 k 的所有满足 nums[k] + nums[i] + nums[j] == 0 的 i,j 组合:
- nums[k] > 0 时直接break跳出:因为 nums[j] >= nums[i] >= nums[k] > 0,即 33 个数字都大于 00 ,在此固定指针 k 之后不可能再找到结果了。
- 当 k > 0且nums[k] == nums[k - 1]时即跳过此元素nums[k]:因为已经将 nums[k - 1] 的所有组合加入到结果中,本次双指针搜索只会得到重复组合。
- i,j 分设在数组索引 (k, len(nums))(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],防止记录到重复组合。
复杂度分析:
时间复杂度 O(N^2)O(N
利用hashset去重
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Set<List<Integer>> list = new HashSet<>();
Arrays.sort(nums);
for(int i= 0; i < nums.length; i++){
int l = i+1;
int r = nums.length-1;
int diff = 0 - nums[i];
while( l <r){
int sum = ( nums[l] + nums[r]);
if( sum == diff){
List<Integer> temp = new ArrayList<>();
temp.add(nums[i]);
temp.add( nums[l]);
temp.add(nums[r]);
list.add(temp);
l++;
r--;
}else if ( sum > diff){
r--;
}else{
l++;
}
}
}
return new ArrayList<>(list);
}
}
转载请注明来源,欢迎对文章中的引用来源进行考证,欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论,也可以邮件至 mym_74@163.com