描述:给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解答:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result=new ArrayList<>();
if(nums.length<3)
return result;
Arrays.sort(nums);
if(nums[0]>0)
return result;
int right=0,left=0;
for(int i=0;i<nums.length-2;i++)
{
if(i>0&&nums[i]==nums[i-1]) continue; //bravo
left=i+1;
right=nums.length-1;
while(left<right)
{
if(nums[left]+nums[right]==-nums[i])
{
List<Integer> t=new ArrayList<>();
t.add(nums[i]);
t.add(nums[left]);
t.add(nums[right]);
result.add(t);
while((left<right-1)&&(nums[left]==nums[left+1]))
left++;
while((left<right-1)&&(nums[right]==nums[right-1]))
right--;
left++;
right--;
}
else if(nums[left]+nums[right]>-nums[i])
{
while((left<right-1)&&(nums[right]==nums[right-1]))
right--;
right--;
}
else
{
while((left<right-1)&&(nums[left]==nums[left+1]))
left++;
left++;
}
}
}
return result;
}
}
思路:
1. 双指针
三数之和为0,即找到三个数满足a+b+c=0。那么可以转换成a+b=-c,即将三数问题转换为两数问题。
先将数组排序。然后从左向右遍历数组,将每一次遍历的数当作c,然后在当前遍历中对这个数之后的数列使用双指针来找a和b满足a+b=-c。
双指针得确定指针的移动规则:
if(nums[左指针]+nums[右指针]=-c)
{
左指针加一;
右指针减一;
}
else if(nums[左指针]+nums[右指针]>-c)
右指针减一; //这是为了让结果逼近-c,下面同理
else if(nums[左指针]+nums[右指针]<-c)
左指针加一;
2. 去重
由于答案中要求不能出现重复的三元组,所以光有了指针的移动规则还不够,去重也是很重要的一部分。
对于c的去重:
if(i>0&&nums[i]==nums[i-1]) continue;
这行代码首先是对a+b=-c中的c去重。每一次遍历向前找是否有相同的数。我之前的思路是向后找相同以去重,即if(nums[i]==nums[i+1]) continue;
但这就大错特错了,因为向后找的话,当遍历遇到一段由相等数字组成的数列段时,会直接从这一段的最后一个数字开始遍历。而向前找相同就能保证遇到一段由相等数字组成的数列段时,能使这一段的第一个数字被遍历到。
对于a和b的去重:
while((left<right-1)&&(nums[left]==nums[left+1]))
left++;
while((left<right-1)&&(nums[right]==nums[right-1]))
right--;
a+b=-c中a和b的去重。首先明确,要对一个数进行去重的基本条件是,我不需要这个数了,那么在nums[left]+nums[right]==-nums[i]
条件下,由于答案不能是重复的三原组,因此nums[left]
和nums[right]
都是不再需要的数,所以对nums[left]
和nums[right]
进行去重。在nums[left]+nums[right]>-nums[i]
条件下,由于此时要将nums[left]+nums[right]
逼近-nums[i]
,因此左指针加一,即不需要nums[left]
,所以对nums[left]
去重。同理nums[left]+nums[right]<-nums[i]
条件下的去重。