力扣解题思路:15. 三数之和 纠错记录

15. 三数之和

思路:在这里插入图片描述
这一题数组多次做错,所以记录一下。思路很简单,固定一个数,再用双指针找另外两个数:

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> res = new ArrayList<>();
    if(nums.length < 3) return res;
    for(int i=0;i<=nums.length-3;i++){
        int sum = -nums[i];
        int l = i+1;
        int r = nums.length-1;
        while(l < r){
            if(nums[l]+nums[r] == sum){
                res.add(Arrays.asList(nums[i],nums[l],nums[r]));
                l++;
                r--;
            }else if(nums[l]+nums[r] > sum){
                r--;
            }else{
                l++;
            }
        }
    }
    return res;
}

然后就错了,测试用例中就可以看出,数组中有重复数字,并且题目要求要不重复的元组,所以需要添加去重操作,因此可以添加上这一句 if(i != 0 && nums[i] == nums[i-1]) continue :

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> res = new ArrayList<>();
    if(nums.length < 3) return res;
    for(int i=0;i<=nums.length-3;i++){
        if(i != 0 && nums[i] == nums[i-1]) continue;
        int sum = -nums[i];
        int l = i+1;
        int r = nums.length-1;
        while(l < r){
            if(nums[l]+nums[r] == sum){
                res.add(Arrays.asList(nums[i],nums[l],nums[r]));
                l++;
                r--;
            }else if(nums[l]+nums[r] > sum){
                r--;
            }else{
                l++;
            }
        }
    }
    return res;
}

但是这样仍然不对,因为这样只保证了三元组的第一个元素不会重复,但后面两个元素是可能被重复计入的,所以在双指针中也需要相应的去重操作。

因为数组排序过,所以相同元素都聚集在一起,我们遇到相同元素直接跳过即可,所以加上这样一句:

while(nums[l] == nums[l-1] && l<r) l++;
while(nums[r] == nums[r+1] && r>l) r--;

注意r>l这个条件是必须的,比如【-4,2,2】最后两个元素相同是可以的:

        while(l < r){
            if(nums[l]+nums[r] == sum){
                res.add(Arrays.asList(nums[i],nums[l],nums[r]));
                l++;
                r--;
                while(nums[l] == nums[l-1] && l<r) l++;
                while(nums[r] == nums[r+1] && r>l) r--;
            }else if(nums[l]+nums[r] > sum){
                r--;
                while(nums[r] == nums[r+1] && r>l) r--;
            }else{
                l++;
                while(nums[l] == nums[l-1] && l<r) l++;
            }
        }

完整代码如下:

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> res = new ArrayList<>();
    if(nums.length < 3) return res;
    for(int i=0;i<=nums.length-3;i++){
        if(i != 0 && nums[i] == nums[i-1]) continue;
        int sum = -nums[i];
        int l = i+1;
        int r = nums.length-1;
        while(l < r){
            if(nums[l]+nums[r] == sum){
                res.add(Arrays.asList(nums[i],nums[l],nums[r]));
                l++;
                r--;
                while(nums[l] == nums[l-1] && l<r) l++;
                while(nums[r] == nums[r+1] && r>l) r--;
            }else if(nums[l]+nums[r] > sum){
                r--;
                while(nums[r] == nums[r+1] && r>l) r--;
            }else{
                l++;
                while(nums[l] == nums[l-1] && l<r) l++;
            }
        }
        return res;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值