三数之和—双指针解法
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
仔细阅读题目,要去寻找一个数组中的三个数和为0的三元组,并且答案中不能重复。以前做过进阶版的两数之和,采用的是双指针方法,但是前提是数组有序,在比较和与目标值之间,同时移动指针来达到减低复杂度的方法。对于这个题,我们先对数组进行排序,并且三个数的和为0,那么必定有负数存在,因为三个数都为正数的话和不可能为0,并且因为要答案中没有重复的,如果遍历到和前一个元素相同的元素时,应该直接跳过,直到遍历到不同的元素。
总的来说,就是先把数组排好序,再固定一个元素,在这个元素之后的元素中寻找剩下两个满足条件的数。
代码如下:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
//如果数组为空或者数组长度小于3,都是不满足条件的。
if(nums == null || nums.length <= 2){
return list;
}
Arrays.sort(nums);
for(int i = 0; i < nums.length - 2; i++){
//如果第一个数大于0 ,那么后面的数都大于0,不可能出现三个正数相加等于0 的情况
if(nums[i] > 0){
break;
}
//去掉重复情况,如果当前的这个元素的值和前一个元素的值相同,那么久直接跳过
if(i > 0 && nums[i] == nums[i - 1]){
continue;
}
//记录当前目标值,取相反数
int stand = -nums[i];
//定义两个指针,头指针和尾指针
int j= i + 1;
int k = nums.length - 1;
//终止条件
while(j < k){
//判断两个元素的和是否与目标值相等
if(nums[j] + nums[k] == stand){
//如果相等,就讲元素添加进列表里
list.add(new ArrayList<>(Arrays.asList(nums[i],nums[j],nums[k])));
//移动指针
j++;
k--;
//判断当前元素和原来的元素是否相同
while(j < k && nums[j] == nums[j - 1]){
j++;
}
while(j < k && nums[k] == nums[k + 1]){
k--;
}
}else if(nums[j] + nums[k] < stand){ //如果比目标小,头指针向前移
j++;
}else{ //否则尾指针向后移
k--;
}
}
}
return list;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。