给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
注意事项
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组。
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
(-1, -1, 2)
/*
这道题让我们求三数之和,比之前那道Two Sum 两数之和要复杂一些,我们还是要首先对原数组进行排序,然后开始遍历排序后的数组,这里注意不是遍历到最后一个停止,而是到倒数第三个就可以了,然后我们还要加上重复就跳过的处理,对于遍历到的数,我们用0减去这个数得到一个sum,我们只需要再之后找到两个数之和等于sum即可,这样一来问题又转化为了求two sum,这时候我们一次扫描,找到了等于sum的两数后,加上当前遍历到的数字,按顺序存入结果中即可,然后还要注意跳过重复数字。
或者我们也可以利用set的不能包含重复项的特点来防止重复项的产生,
思路:
首先,将数组按从小到大的排序,然后从头挑选一个元素,接着使用首尾两个指针来挑选后两个元素(被选定的就不要了);
*/
public class Solution{
public ArrayList<ArrayList<Integer>> threeSum(int[] nums){
ArrayList<ArrayList<Integer>> lists = new ArrayList<>();
if(nums ==null || nums.length ==0)return lists;
Arrays.sort(nums);
for(int i=0;i<nums.length-2;i++){
//第一个数大于0,因为是从小到大排了序的,所以以下就不可能了。
去除重复结果
if(i>0 && nums[i] ==nums[i-1]){
continue;
}
int left =i+1;
int right = nums.length-1;
int target = -nums[i];
//两指针向中间靠拢找结果
while(left<right){
if(nums[left] + nums[right] <target){
left++;
}else if(nums[left] + nums[right] >target){
right--;
}else{
ArrayList<Integer> list = new ArrayList<>();
list.add(-target);
list.add(nums[left]);
list.add(nums[right]);
lists.add(list);
//k可能有多个集合
left++;
right--;
// //避免重复
while(left<right&&nums[left] == nums[left-1]){
left++;
}
while(left<right && nums[right] == nums[right+1]){
right--;
}
}
}
}
return lists;
}
}