给定一个包含 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>> list = new ArrayList();
Arrays.sort(nums);
int a, b, c;
for (int i = 0; i < nums.length - 2; ) {
a = nums[i];
int j = i + 1;
int k = nums.length - 1;
while (j < k) {
b = nums[j];
c = nums[k];
if (a + b + c == 0) {
List<Integer> item = new ArrayList<>();
item.add(a);
item.add(b);
item.add(c);
list.add(item);
j++;
k--;
while (j < k && nums[j] == nums[j - 1]) {
j++;
}
while (j < k && nums[k + 1] == nums[k]) {
k--;
}
} else if (a + b + c > 0) {
k--;
while (j < k && nums[k + 1] == nums[k]) {
k--;
}
} else {
j++;
while (j < k && nums[j] == nums[j - 1]) {
j++;
}
}
}
++i;
while (i < nums.length - 2 && nums[i] == nums[i - 1]) {
i++;
}
}
return list;
}
}