题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2] ]
思路
两层for循环就可以确定 a 和b 的数值了,可以使用哈希法来确定 0-(a+b) 是否在 数组里出现过,其实这个思路是正确的,但是我们有一个非常棘手的问题,就是题目中说的不可以包含重复的三元组。
把符合条件的三元组放进vector中,然后在去去重,这样是非常费时的,很容易超时,也是这道题目通过率如此之低的根源所在。
这道题目使用哈希法并不十分合适
双指针
首先将数组排序,然后有一层for循环,i从下表0的地方开始,同时定一个下表left 定义在i+1的位置上,定义下表right 在数组结尾的位置上。
在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i] b = nums[left] c = nums[right]。
接下来如何移动left 和right???
如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下表就应该向左移动,这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明 此时三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
工具类 Arrays.asList()把数组转换成集合
代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
//int[] res = new int[3];
List<List<Integer>> result = new ArrayList<>();
Arrays.sort(nums);
for(int i = 0; i < nums.length; i++){
if(nums[i] > 0){
return result;
}
// 去重
if( i > 0 && nums[i] == nums[i - 1] ){
continue;
}
int left = i + 1;
int right = nums.length - 1;
while(left < right){
int sum = nums[i]+nums[left]+nums[right];
if(sum > 0){
right--;
}else if(sum < 0 ){
left++;
}else{
result.add(Arrays.asList(nums[i],nums[left],nums[right]));
while(left < right && nums[right] == nums[right-1]) right--; // 去重
while(left < right && nums[left] == nums[left+1]) left++; // 去重
left++;
right--;
}
}
}
return result;
}
}