15. 三数之和
思路:
这一题数组多次做错,所以记录一下。思路很简单,固定一个数,再用双指针找另外两个数:
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
if(nums.length < 3) return res;
for(int i=0;i<=nums.length-3;i++){
int sum = -nums[i];
int l = i+1;
int r = nums.length-1;
while(l < r){
if(nums[l]+nums[r] == sum){
res.add(Arrays.asList(nums[i],nums[l],nums[r]));
l++;
r--;
}else if(nums[l]+nums[r] > sum){
r--;
}else{
l++;
}
}
}
return res;
}
然后就错了,测试用例中就可以看出,数组中有重复数字,并且题目要求要不重复的元组,所以需要添加去重操作,因此可以添加上这一句 if(i != 0 && nums[i] == nums[i-1]) continue :
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
if(nums.length < 3) return res;
for(int i=0;i<=nums.length-3;i++){
if(i != 0 && nums[i] == nums[i-1]) continue;
int sum = -nums[i];
int l = i+1;
int r = nums.length-1;
while(l < r){
if(nums[l]+nums[r] == sum){
res.add(Arrays.asList(nums[i],nums[l],nums[r]));
l++;
r--;
}else if(nums[l]+nums[r] > sum){
r--;
}else{
l++;
}
}
}
return res;
}
但是这样仍然不对,因为这样只保证了三元组的第一个元素不会重复,但后面两个元素是可能被重复计入的,所以在双指针中也需要相应的去重操作。
因为数组排序过,所以相同元素都聚集在一起,我们遇到相同元素直接跳过即可,所以加上这样一句:
while(nums[l] == nums[l-1] && l<r) l++;
while(nums[r] == nums[r+1] && r>l) r--;
注意r>l这个条件是必须的,比如【-4,2,2】最后两个元素相同是可以的:
while(l < r){
if(nums[l]+nums[r] == sum){
res.add(Arrays.asList(nums[i],nums[l],nums[r]));
l++;
r--;
while(nums[l] == nums[l-1] && l<r) l++;
while(nums[r] == nums[r+1] && r>l) r--;
}else if(nums[l]+nums[r] > sum){
r--;
while(nums[r] == nums[r+1] && r>l) r--;
}else{
l++;
while(nums[l] == nums[l-1] && l<r) l++;
}
}
完整代码如下:
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
if(nums.length < 3) return res;
for(int i=0;i<=nums.length-3;i++){
if(i != 0 && nums[i] == nums[i-1]) continue;
int sum = -nums[i];
int l = i+1;
int r = nums.length-1;
while(l < r){
if(nums[l]+nums[r] == sum){
res.add(Arrays.asList(nums[i],nums[l],nums[r]));
l++;
r--;
while(nums[l] == nums[l-1] && l<r) l++;
while(nums[r] == nums[r+1] && r>l) r--;
}else if(nums[l]+nums[r] > sum){
r--;
while(nums[r] == nums[r+1] && r>l) r--;
}else{
l++;
while(nums[l] == nums[l-1] && l<r) l++;
}
}
return res;
}
}