题目介绍
题目
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题代码
思路:
跟两数之和不同的是,三数之后要输出的是不同三元组的集合。因此,我们考虑先将nums进行排序,将nums[i]作为第一个加数,从i+1到nums.length-1之间初始化两个指针left,right,为了避免有重复的情况,当nums[i]==nums[i-1],说明有重复的情况,开始下一个循环。如果num[i]+num[left]+num[right]>0,说明加多了,让right–,如果num[i]+num[left]+num[right]<0,说明加少了,让left++,如果等于0,说明符合条件,将这一组解加到集合中,这是也应该避免第二个加数和第三个加数重复的情况。
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList<List<Integer>>();
Arrays.sort(nums);
for (int i = 0; i < nums.length - 1; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1;
int right = nums.length - 1;
while (left < right) {
if (nums[left] + nums[right] + nums[i] > 0) {
right--;
} else if (nums[left] + nums[right] + nums[i] < 0) {
left++;
} else {
list.add(Arrays.asList(nums[left], nums[right], nums[i]));
left++;
right--;
while (left < right && nums[left] == nums[left - 1]) {
left++;
}
while (left < right && nums[right] == nums[right + 1]) {
right--;
}
}
}
}
return list;
}