https://leetcode-cn.com/problems/3sum/
给你一个包含 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]
输出:[]
方法
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
if (nums == null || nums.length <= 2) return ans;
Arrays.sort(nums); // O(nlogn)
for(int i=0;i<nums.length-2;i++){
if(nums[i]>0) break;
if(i>0&&nums[i]==nums[i-1]) continue; // 去除重复target
int target=-nums[i];
int pre=i+1;
int last=nums.length-1;
while(pre<last){
if(nums[pre]+nums[last]==target){
ans.add(new ArrayList<Integer>(Arrays.asList(nums[pre],nums[last],-target)));
pre++;
last--;
while(pre<last&&nums[pre]==nums[pre-1]) pre++; // 去重数组
while(pre<last&&nums[last]==nums[last+1]) last--; // 去重数组
}
else if(nums[pre]+nums[last]<target){
pre++;
}
else{
last--;
}
}
}
return ans;
}
}
结果