题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法
本题可以先找一个基准数,然后去找另外两个数,另外两个数可以使用双指针法,然后注意剪枝。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
int l = nums.length;
List<List<Integer>> list = new ArrayList<List<Integer>>();
for(int i = 0; i < l - 2; i++){
// 最小值都大于0了,后面两个数更大,所以这个数无法找到另外两个数和它匹配
// 而且因为nums[i]后面的数都比nums[i]更大,而且后面的也大于0,同理也找不到
if(nums[i] > 0){
break;
}
// 如果nums[i]和nums[i-1]相等,所以以nums[i]为基准的数,在之前的循环中以全部找到了
if(i > 0 && nums[i] == nums[i-1]){
continue;
}
int left = i + 1;
int right = l - 1;
while(left < right){
int sum = nums[i] + nums[left] + nums[right];
if(sum == 0){
list.add(Arrays.asList(nums[i],nums[left],nums[right]));
// 这个也是为了剪枝,保证right不会一样
while(left < right && nums[right] == nums[--right]);
}else if(sum > 0){
//大于0,表示右边的数太大了,右坐标左移
right--;
}else{
//大于0,表示左边的数太小了,左坐标右移
left++;
}
}
}
return list;
}
}