题目:输入一个数组,找出数组中所有的三个数和为0的组合
分析:
1.先排序,对数组进行排序。
2.然后使用双指针法。首先固定一个元素,循环遍历固定元素,然后从固定元素的下一个开始定义 左右指针,如果三者加和大于0,则右指针向左移,小于0,则左指针向右移,如果固定的元素直接就大于0了,因为刚开始数组排好序的,如果此时固定元素就大于0了,那后面元素就都大于0,所以后面的三数组合之和不可能在小于0,那就说明没必要再次寻找了。
需要注意的点:1.记得要先排序,排好序的数组更方便用于去重,因为重复元素都排列在一起了。2.要注意左右指针的去重操作,不论是大于0,还是小于0,甚至是等于0,在移动指针时都要注意去重,可以大大加快程序的运行速度。
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
// 第一个元素为k,固定,然后从第二个元素开始,到length-1个元素结束,一共length-2个元素
for(int k = 0; k < nums.length - 2; k++){
// 因为该数组排好序了,所以此时如果k都大于0,那么后面的元素肯定大于0,证明搜索结束
if(nums[k] > 0)
break;
// 证明数组中有数据重复,跳过重复数据
if(k > 0 && nums[k] == nums[k - 1])
continue;
// 从固定元素的下一个开始查找
int i = k + 1, j = nums.length - 1;
while(i < j){
int sum = nums[k] + nums[i] + nums[j];
if(sum < 0){
while(i < j && nums[i] == nums[++i]);
} else if (sum > 0) {
while(i < j && nums[j] == nums[--j]);
} else {
res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
while(i < j && nums[i] == nums[++i]);
while(i < j && nums[j] == nums[--j]);
}
}
}
return res;
}