LeetCode Hot 100 --- 三数之和 （java详解）

Posted 2021-09-20 小样5411

题目

代码

用的是双指针，双指针在需要遍历的情况下经常被使用，且效率不错，下面代码做了详细注释

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if(nums.length < 3){
            return res;
        }
        //如何找到三个和为0的数字？
        //先给数组排序，这样的话，后面可以当nums[i]>0时，直接break，因为有序，之后三元组肯定更>0
        Arrays.sort(nums);
        int len = nums.length;
        //遍历整个数组找三元组
        for(int i = 0 ; i < len ; i++){
            //每次先判断当前位置的nums[i]是否>0，是则break
            if(nums[i] > 0) break;
            //每次判断相邻nums[i]是否重复,重复则continue，执行下次循环
            if(i > 0 && nums[i] == nums[i-1]) continue;
            int left = i+1;//每次定义双指针，left和right在i+1到len-1范围内找符合条件的数
            int right = len - 1;
            //left >= right ，即相遇跳出循环
            while(left < right){
                int sum = nums[i] + nums[left] + nums[right];
                if(sum == 0){
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    //添加后看是否有重复，有重复也直接排除
                    while(left < right && nums[left] == nums[left + 1]) left++;//去重
                    while(left < right && nums[right] == nums[right - 1]) right--;//去重
                    //经过上面，不重复才使用，nums[left] != nums[left + 1]且nums[right] != nums[right - 1]，下面才会走到left+1和right-1位置
                    left++;
                    right--;
                }else if(sum > 0){
                    //大了则要缩小
                    right--;
                }else{
                    //小了则要变大
                    left++;
                }

            }
        }
        return res;
    }
}