0双指针中等 leetcode15. 三数之和

排序+双指针
排序,方便区分正负数。三个数和为0必定有一个是小于等于0。
设三个变量,i在小于等于0的区间里遍历,j初始位置是i+1,k初始位置是nums.length-1
i在小于等于0的区间里遍历,每在一个位置,都让j,k遍历数组直到j不小于k.i在遍历时还要注意去重。
当三个数的和小于0,j++
当三个数的和大于0,k–
当三个数的和等于0,保存这三个数,j++、k–(注意nums[k-1] == nums[k]时,要跳过k,避免重复计算)
有两个地方需要去重:

  1. i在小于等于0的区间里遍历时,当nums[i]与nums[i-1]相同时要跳过i。
  2. j、k的位置更新时,与i一样注意避免计算相同的nums[j]和nums[k]。
import java.util.ArrayList;
import java.util.Arrays;
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if(nums.length < 3){
            return res;
        }
        Arrays.sort(nums);
        for(int i = 0; i < nums.length-2; i++){
            if(nums[i] > 0){
                break;
            }
            if(i > 0 && nums[i] == nums[i-1]){//去重
                continue;
            }
            int j = i + 1;
            int k = nums.length - 1;
            while(j < k){
                int sum = nums[i] + nums[j] + nums[k];
                if(sum < 0){
                    j++;
                }else if(sum > 0){
                    k--;
                }else{
                    List<Integer> li = new ArrayList<>();
                    li.add(nums[i]);
                    li.add(nums[j]);
                    li.add(nums[k]);
                    res.add(li);
                    while(j < k && nums[k-1] == nums[k]){//去重
                        k--;
                    }
                    while(j < k && nums[j+1] == nums[j]){//去重
                        j++;
                    }
                    k--;//很重要,否则超时,还没明白为什么超时。
                    j++;
                }
            }
        }
        return res;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值