给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ,找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
分析:
用一个数组,两个指针索引left,right. left从当前位置i的下一个位置,right从最后一个位置查找,若三个位置所对应的值之和等于0,则将它添加到列表中。每次从左向右(或从右向左)找时都要找第一个与之前处理的数不同的数的下标,找完符合条件的数后指向下一个要处理的数。
mport java.util.*;
//使得a+b+c=0,输出不重复数组
public class findarray {
public static List<List<Integer>> findarray(int[] array) { //一个数组,两个指针
List<List<Integer>> ret = new ArrayList<>();
if (array == null || array.length < 3) {
return ret;
}
Arrays.sort(array);
for (int i = 0; i < array.length - 2; ) {
int left=i+1;
int right=array.length-1;
while (left < right) {
if ( array[i] + array[left] + array[right] == 0) {
ret.add(Arrays.asList(new Integer[]{array[i], array[left], array[right]}));
//移动到下一个位置,找下一组解
left++;
right--;
while (left < right && array[left] == array[left - 1]) { //从左向右找第一个与之前处理的数不同的数的下标
left++;
}
while (left < right && array[right] == array[right + 1]) {
right--;//从右向左找第一个与之前处理的数不同的数的下标
}
} else if ((array[i] + array[left] + array[right]) < 0) {
left++;
while (left < right && array[left] == array[left - 1]) { //从左向右找第一个与之前处理的数不同的数的下标
left++;
}
} else {
right--;
while (left < right && array[right] == array[right + 1]) {
right--;//从右向左找第一个与之前处理的数不同的数的下标
}
}
}
i++;//指向下一个要处理的数
while(i<array.length-2&&array[i]==array[i-1]){
i++;
}
}
return ret;
}
public static void main(String[] args) {
findarray f=new findarray();
int[] array={1,4,5,6,7,6,5,3,3-1,-2,-4,-6,6,0};
System.out.println(f.findarray(array));
}
}
运行结果