题目:给定一个包含 n 个整数的数组 S,是否存在属于 S 的三个元素 a,b,c 使得 a + b + c = 0 ?找出所有不重复的三个元素组合使三个数的和为零。
注意:结果不能包括重复的三个数的组合。
例如, 给定数组 S = [-1, 0, 1, 2, -1, -4],
一个结果集合为: [ [-1, 0, 1], [-1, -1, 2] ]
要点:
要点一:先排序,利用前后指针去寻找等于 0-(1)的组合
要点二:不重复的的组合,在遍历时要过滤掉相同元素。三个指针都需要过滤。一共四个地方,外层循环,内层循环3个地方
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
if(nums == null || nums.length==0) {
return result ;
}
nums = quickSort(nums, 0, nums.length-1);
for(int i=0; i < nums.length; i++) {
// 从左向右找第一个与之前处理的数不同的数的下标
if (i > 0 && i < nums.length - 2 && nums[i] == nums[i - 1]) { //(1)
continue;
}
int sum = 0 - nums[i];
int j = i + 1;
int k = nums.length -1;
while(j < k) {
if(nums[j] + nums[k] > sum) {
k--;
while (j < k && nums[k] == nums[k + 1]) { //(2)
k--;
}
} else if (nums[j] + nums[k] < sum) {
j++;
while (j < k && nums[j] == nums[j - 1]) { //(3)
j++;
}
} else {
List<Integer> list = new ArrayList<>();
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[k]);
result.add(list);
k--;
j++;
while (j < k && nums[j] == nums[j - 1]) { //(4)
j++;
}
while (j < k && nums[k] == nums[k + 1]) { //(4)
k--;
}
}
}
}
return result;
}
int[] quickSort(int a[], int start, int end) {
int mid = partion(a, start, end);
if(mid - 1 > start ) {
quickSort(a, start, mid - 1);
}
if(mid + 1 < end ) {
quickSort(a, mid + 1, end);
}
return a;
}
int partion(int a[], int start, int end) {
int temp = a[start];
while(start < end) {
while(a[end] >= temp && start < end) {
end--;
}
a[start] = a[end];
while(a[start] <= temp && start < end) {
start++;
}
a[end] = a[start];
}
a[start] = temp;
return start;
}
}