给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
我自己使用的一个三层for循环,但是超时了;
public static List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> list = new ArrayList<>();
int len = nums.length;
for (int i = 0; i < nums.length; i++) {
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
for (int j = i + 1; j < nums.length; j++) {
if (nums[j] == nums[j - 1] && j > i + 1) {
continue;
}
for (int k = j + 1; k < nums.length; k++) {
if (nums[k] == nums[k - 1] && k > j + 1) {
continue;
}
int temp = nums[i] + nums[j] + nums[k];
if (temp == 0) {
List<Integer> list1 = new ArrayList<>();
list1.add(nums[i]);
list1.add(nums[j]);
list1.add(nums[k]);
list.add(list1);
}
}
}
}
return list;
}
第二中方法可以通过,我是参考别人的:
public static List<List<Integer>> threeSum(int[] nums) {
int len = nums.length;
List<List<Integer>> list = new ArrayList<List<Integer>>();
if (nums == null || nums.length < 3) {
return list;
}
Arrays.sort(nums);
for (int i = 0; i < len; i++) {
if (i > 0 && nums[i - 1] == nums[i]) {
continue;
}
int begin = i + 1;
int end = len - 1;
while (begin < end) {
int temp = nums[i] + nums[begin] + nums[end];
if (temp == 0) {
List<Integer> list1 = new ArrayList<>();
list1.add(nums[i]);
list1.add(nums[begin]);
list1.add(nums[end]);
list.add(list1);
end--;
begin++;
while (begin < end && nums[begin] == nums[begin - 1]) {
begin++;
}
while (begin < end && nums[end] == nums[end + 1]) {
end--;
}
} else if (temp > 0) {
end--;
} else {
begin++;
}
}
}
return list;
}
}