给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0?
请你找出所有满足条件且不重复的三元组。 注意:答案中不可以包含重复的三元组。
示例: 给定数组 nums = [-1, 0, 1,2, -1, -4],
满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
一般这种找值的都是滑动窗口解决的(我认为奥),而滑动窗口要建立在排序的基础上,代入了一下滑动窗口的思维确实可以解答,
比如上述的例题,先排序得[-4,-1,-1,0,1,2],
一般滑动窗口是两个数求和的,这个求3个数只需用循环先提出一个数,再对剩余的数取值即可,例如提出-4,得[-1,-1,0,1,2],对剩余的数组内求和4即可,
定义left和right,和大于4则right–,小于则left++,思路是很清晰的,但是就是有一点不能包含重复的三元组,则必须在每次移动的时候判断是否重复
代码如下:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> rs = new ArrayList<List<Integer>>();
if(nums.length < 3) return rs;
Arrays.sort(nums);
int i = 0;
while(i < nums.length-1){
int right = nums.length - 1;
for(int left = i + 1;left < right;){
if(nums[i] + nums[left] + nums[right] == 0){
List<Integer> list = new ArrayList<Integer>();
list.add(nums[i]);
list.add(nums[left]);
list.add(nums[right]);
rs.add(list);
left++;
while(left < right && nums[left] == nums[left - 1]){
left++;
}
}else if(nums[i] + nums[left] + nums[right] > 0){
right--;
while(left < right && nums[right] == nums[right + 1]){
right--;
}
}else{
left++;
while(left < right && nums[left] == nums[left - 1]){
left++;
}
}
}
i++;
while(i < nums.length && nums[i] == nums[i - 1]){
i++;
}
}
return rs;
}
}
官方的代码思路一致,但是代码对于排重复不一样,还是官方的牛一点。
代码如下:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int n = nums.length;
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<List<Integer>>();
// 枚举 a
for (int first = 0; first < n; ++first) {
// 需要和上一次枚举的数不相同
if (first > 0 && nums[first] == nums[first - 1]) {
continue;
}
// c 对应的指针初始指向数组的最右端
int third = n - 1;
int target = -nums[first];
// 枚举 b
for (int second = first + 1; second < n; ++second) {
// 需要和上一次枚举的数不相同
if (second > first + 1 && nums[second] == nums[second - 1]) {
continue;
}
// 需要保证 b 的指针在 c 的指针的左侧
while (second < third && nums[second] + nums[third] > target) {
--third;
}
// 如果指针重合,随着 b 后续的增加
// 就不会有满足 a+b+c=0 并且 b<c 的 c 了,可以退出循环
if (second == third) {
break;
}
if (nums[second] + nums[third] == target) {
List<Integer> list = new ArrayList<Integer>();
list.add(nums[first]);
list.add(nums[second]);
list.add(nums[third]);
ans.add(list);
}
}
}
return ans;
}
}