题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
解题思路
直接用三层循环枚举每一个数肯定会TLE。当时想的排个序,然后开个二重循环,用二分查找来优化,这样时间复杂度是O(n 2 ^{2} 2log(n)),还是会TLE。
正解是排个序,要利用序列的有序性,先枚举第一个数 nums[i]
,然后用两个指针 left
和 right
分别从数组剩余部分的左右两段向中间靠拢, 直到二者相遇。当 sum==0
的时候,三个指针对应的数即为所求,当 sum<0
的时候,则表示 left
指针对应的数太小,需要将其向右移动,当 sum>0
的时候,则表示 right
指针对应的数太大,需要将其向左移动。这样整个过程算下来时间复杂度是O(n
2
^{2}
2)。由于题目中要求去除重复的元组,那么在整个过程中,指针要跳过相同元素。
代码
被凎碎了。。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
int length = nums.length;
if (length < 3) return ans;
Arrays.sort(nums);
for (int i = 0; i < length; i++) {
if (nums[i] > 0) break;
if (i != 0 && nums[i] == nums[i - 1]) continue;
int left = i + 1, right = length - 1;
while (left < right) {
int sum = nums[i] + nums[left] + nums[right];
if (sum == 0) {
List<Integer> tmp = new ArrayList<>();
tmp.add(nums[i]);
tmp.add(nums[left]);
tmp.add(nums[right]);
ans.add(tmp);
while (left < right) {
if (nums[left + 1] == nums[left])
left++;
else break;
}
while (left < right) {
if (nums[right - 1] == nums[right])
right--;
else break;
}
left++;
right--;
} else if (sum < 0) {
left++;
} else if (sum > 0) {
right--;
}
}
}
return ans;
}
}