描述:给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
思路: 三数之和可以看成是两个数的和等于另一个数的负数,假定一个数的负数,然后寻找两数之和等于这个数,为了提高效率,我们从两边出发找这两个数,由于假定的数可能为正,可能为负,因此我们先对这个数组进行排序,然后从两端寻找两数之和,此时数组两端一大一小,如果两数之和等于原先假定的和的话,就找到一个,然后继续寻找。如果两数之和小于假定的数,那么就是数小了,从左边小的数增加,如果数大了,那么从右端数减小数。
代码如下:
public static List<List<Integer>> threeSum1(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if(nums == null || nums.length < 3)
return res;
Arrays.sort(nums);
for(int i = 0; i<nums.length-2; i++){
if(i == 0 || (i>0 && nums[i] != nums[i-1])){
int lo = i+1, hi = nums.length-1, sum = -nums[i];
while(lo < hi){
if((nums[lo] + nums[hi]) == sum){
res.add(Arrays.asList(nums[i],nums[lo],nums[hi]));
while((lo < hi) && (nums[lo] == nums[lo+1])) lo++;
while((lo < hi) && (nums[hi] == nums[hi-1])) hi--;
lo++;
hi--;
}else if((nums[lo] + nums[hi]) < sum) lo++;
else hi--;
}
}
}
return res;
}