import java.util.ArrayList;
import java.util.Arrays;
public class ThreeSum {
public static void main(String[] args) {
int[] array = {0,0,0,0};
ArrayList<ArrayList<Integer>> result = threeSum(array);
for(ArrayList<Integer> item : result) {
for(int a : item) {
System.out.print(a + " ");
}
System.out.println();
}
}
/**
* 基本思想:我们需要在数组中选取三个相加等于0的数,有很多中解法。
* 解法一:暴力破解,将该数组中三个元素的组合全部找出来(使用一个递归选取一个元素删除该元素再选)
* 然后从从中筛选出满足我们条件的三元组。
* 解法二:
* 步骤如下:
* ①对数组排序,这样方便我们利用数组中数的顺序来解决该问题
* ②从数组的左边找到一个不大于0的数key,由于数组顺序排列,如果第一个数大于0后面两个数也大于0加起来不可能为0
* ③定义两个变量left,right分别指向key的下一个元素和数组的最后一个变量
* ④如果left、right和key的和大于0那么我们将right向左移动(变小),如果小于0我们就将left右移,直到找到符合我们条件的元素
* 或者left和right指向同一个元素(不能找到满足条件的元素)
* ⑤循环执行2,3,4步直到下一个元素大于0或者遍历到数组的最后两个元素
* 注意:
* ①我们找key时需要注意数组中元素可能重复每个数我们只取一遍所以我们只遍历一遍,我们只对第一个key做如上操作
* 原因是(-2,-2,4)(0,0,0)这样的三元组可能由两个相同的元素组成key
* ②一样的如果right有多个相同的元素我们也只处理第一个,为了避免漏算类似于(-10,5,5)这样的元素
* */
public static ArrayList<ArrayList<Integer>> threeSum(int[] num) {
//排序,不喜欢用Arrays的可以自己写,可以顺带练习一下排序
Arrays.sort(num);
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
for(int i = 0;i < num.length - 2 && num[i] <= 0;i++) {
int point1 = num.length - 1;
//只遍历第一个key
while(0 < i && i < num.length && num[i] == num[i-1]) {
i++;
}
int point2 = i + 1;
while(point1 > point2) {
if(num[point1] + num[point2] > -num[i]) {
point1--;
}else if(num[point1] + num[point2] < -num[i]) {
point2++;
}else {
ArrayList<Integer> item = new ArrayList<Integer>();
item.add(num[i]);
item.add(num[point2]);
item.add(num[point1]);
result.add(item);
//只遍历第一个right
while(num[point1] == num[point1 - 1] && point1 > point2){
point1--;
}
point1--;
}
}
}
return result;
}
}
数组中相加和为0的三元组给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?
最新推荐文章于 2022-11-07 09:00:00 发布