import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class test_15 {
public static void main(String[] args){
int[] nums = {-1, 0, 1, 2, -1, -4};
List<List<Integer>> lists = threeSum(nums);
System.out.println(lists);
}
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
//不满足的情况,返回空
if (nums.length <= 2 || nums == null){
return list;
}
//将数组排序
Arrays.sort(nums);
//a+b+c = sum
//枚举a
for (int i = 0; i < nums.length; i++) {
//如果第一个数a都大于0,说明没有满足的情况
if (nums[i] > 0) break;
//如果a与上一次的a相同的话,跳过本次循环
if (i > 0 && nums[i - 1] == nums[i]){
continue;
}
//c的索引
int k = nums.length - 1;
int j = i + 1;
while(j < k){
int sum = nums[i] + nums[j] + nums[k];
if (sum == 0){
List<Integer> subList = new ArrayList<>();
subList.add(nums[i]);
subList.add(nums[j]);
subList.add(nums[k]);
list.add(subList);
j = j + 1;
k = k -1;
} else if(sum > 0){
int k0 = k - 1;
while (j < k0 && nums[k0] == nums[k]){
k0--;
}
k = k0;
}else {
int j0 = j + 1;
while (j0 < k && nums[j0] == nums[j]){
j0++;
}
j = j0;
}
}
}
return list;
}
}
单层for循环:
1、数组长度小于2或为空的,直接返回空list
2、将数组进行排序Arrays.sort()
3、sum = a + b + c;for循环枚举a,b=a+ 1,c从数组末端开始遍历
如果a > 0 ,那么可以直接结束循环
如果sum > 0 ,需要c左移
如果sum < 0,需要b右移
另解:
遍历a,遍历b,c=nums.length-1;