题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?
请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
解题思路:
先将数组排序,然后循环找出两个数的和,用二分查找找出剩余的那个数。要注意不能重复。
(不过官方题解是直接遍历,但是为什么二分比直接遍历慢?)
Java代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int len = nums.length;
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<List<Integer>>();
for(int i = 0; i < len; i++) {
if(i!=0 && nums[i] == nums[i-1])continue;
for(int j = i+1; j < len; j++) {
if(j!=i+1 && nums[j] == nums[j-1])continue;
int a = -1 * (nums[i] + nums[j]);
int l = j + 1, r = len - 1;
while(l<r){
if(r-l==1) {
if(nums[l] == a) {}
else l = r;
}
int mid = (l+r)>>1;
if(nums[mid] >= a)r = mid;
else l = mid;
}
if(l>j && l<len && nums[l] == a) {
List<Integer>x = new ArrayList<Integer>();
x.add(nums[i]);
x.add(nums[j]);
x.add(a);
ans.add(x);
}
}
}
return ans;
}
}