第18题. 四数之和

2024-01-02 12:30:21

双指针法

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            // 如果nums[i]是负数,那么累计和还会持续地变小
            if (nums[i] > 0 && nums[i] > target)
                break;
            // 第一个位置是i已经遍历过
            if (i > 0 && nums[i] == nums[i - 1])
                continue;
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] + nums[j] > 0 && nums[i] + nums[j] > target)
                    break;
                // 第二个位置是j已经遍历过
                // 为什么是j>i+1而不是j>0 因为现在j的可取值范围是从[i+1,nums.length-1]
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                int left = j + 1;
                int right = nums.length - 1;
                while (left < right) {
                    int total = nums[i] + nums[j] + nums[left] + nums[right];
                    if (total > target)
                        right--;
                    else if (total < target)
                        left++;
                    else {
                        // total==target
                        list.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        while (left < right && nums[left + 1] == nums[left])
                            left++;
                        while (left < right && nums[right - 1] == nums[right])
                            right--;
                        left++;
                        right--;
                    }
                }
            }
        }
        return list;
    }
}

思考

为什么使用双指针法而不是哈希表?因为需要去重
剪枝也是一个重要的问题,target如果是负数,那么还需要判断nums[i]是不是大于0,这种情况才可以剪枝。

文章来源:https://blog.csdn.net/qq_44047415/article/details/135336855
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。