刚做这个题,以为可以用两数之和的思路做。
显然用暴力求解的方法可以参考两数之和。但是好像是重复了。
对于这种三个元素的,一般采用固定一个元素,然后用“双探针”的方法去尝试找解。
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
int len = nums.length;
//这里不能返回null,因为函数返回的参数类型是List
if (nums == null || len < 3) return ans;
//排序默认是升序
Arrays.sort(nums);
//遍历每一个元素,并在后面的元素中寻找两个满足条件的元素
//这里的任何两个三元素元素不能相同,否则的话算是重复的,排序和下面判断取除重复元素的操作实际上是必要的
for (int i = 0; i < len; i ++){
if (nums[i] > 0) break;
if(i > 0 && nums[i] == nums[i-1]) continue; //取除重复的元素,避免出现重复三元组
int L = i + 1;
int R = len - 1;
while (L < R){
int sum = nums[i] + nums[L] + nums[R];
if (sum == 0){
ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
while (L < R && nums[L] == nums[L + 1]) L ++; //去重的时候是在满足sum=0的情况下,剔除所有重复三元组的情况,此时L和R都在变化,所以还是要在while内判断
while (L < R && nums[R] == nums[R - 1]) R --;
L++; //左探针向右探寻
R--; //右探针向左探寻
}
else if (sum < 0) L ++; //sum小了 左探针向右探寻
else if (sum > 0) R --; //sum大了 右探针向左探寻
}
}
return ans;
}