day08-三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
解题思路:
- 特殊情况:数组长度小于3,直接返回空集合
- 排序后:最大值小于0或最小值大于0,直接返回空集合
- 从第一个数开始,一直循环到值大于0的数,从第二位开始如果该数和上次循环的数相同(i>0&&nums[i]==nums[i-1]),表示该数已经遍历过,跳过本次循环。
- 从第3步得到的数的下一位开始直到数组的末尾,用双针法从两端向中间逐步移动,直到两指针位置相同,判断三个数之和,
- 如果等于0,判断双针的下一位是否和当前位相同,如果相同就移动到下一位然后继续判断(剔除重复元素),当双针的下一位都不和当前位相同或者两个指针位置相同时结束循环,记录组合。
- 如果不等于0,大于0,表示三个数的和太大,向左移动右指针来减小三个数的和;小于0,表示三个数的和太小,向右移动左指针来增大三个数的和。
解题代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList();
if(nums.length<3){
return list;
}
int left = 0;
int right = nums.length-1;
Arrays.sort(nums);
if(nums[left]>0||nums[right]<0){
return list;
}
for(int i=0;i<nums.length&&nums[i]<=0;i++){
left = i+1;
right = nums.length-1;
if(i>0&&nums[i]==nums[i-1]){
continue;
}
while(left<right){
if(nums[i]+nums[left]+nums[right]==0){
while(left<right){
if(nums[left]==nums[left+1]){
left++;
}else if(nums[right]==nums[right-1]){
right--;
}else{
break;
}
}
List<Integer> arrayList = new ArrayList();
arrayList.add(nums[i]);
arrayList.add(nums[left]);
arrayList.add(nums[right]);
list.add(arrayList);
}
if(nums[i]+nums[left]+nums[right] > 0){
right--;
}else{
left++;
}
}
}
return list;
}
}
执行效率: