flag
软件学院大三党,每天一道算法题,第十五天
题目介绍
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[-1, 0, 1],
[-1, -1, 2]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
思路
双指针法铺垫: 先将给定 nums 排序,复杂度为 O(NlogN)
双指针法思路: 固定 3 个指针中最左(最小)数字的指针 k,双指针 i,j 分设在数组索引 (k,len(nums)) 两端,通过双指针交替向中间移动,记录对于每个固定指针 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],防止记录到重复组合。
复杂度分析:
时间复杂度 O(N^2):其中固定指针k循环复杂度 O(N)O(N),双指针 i,j 复杂度 O(N)。
空间复杂度 O(1):指针使用常数大小的额外空间。
关键代码
public static List<List<Integer>> threeSum(int[] nums){
Arrays.sort(nums);
List<List<Integer>> result=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;
int j=nums.length-1;
while (i<j){
int sum=nums[k]+nums[i]+nums[j];
if(sum<0){
i++;
while (i<j && nums[i]==nums[i-1]){//跳过重复元素
i++;
}
}
else if(sum>0){
j--;
while (i<j && nums[j]==nums[j+1]){//跳过重复元素
j--;
}
}
else {
List ans=new ArrayList<Integer>();
ans.add(nums[k]);
ans.add(nums[i]);
ans.add((nums[j]));
result.add(ans);
i++;
while (i<j && nums[i]==nums[i-1]){//跳过重复元素
i++;
}
j--;
while (i<j && nums[j]==nums[j+1]){//跳过重复元素
j--;
}
}
}
}
return result;
}