LeetCode18 4Sum

LeetCode18 4Sum

题目描述：

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

答案中不可以包含重复的四元组。

示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

解答：

解法一：排序法加双指针

基本思路是，先排序，两层for循环，nums[i]和nums[j]这会确定一个和， 然后从j+1到len-1处寻找一个两数之和等于target-sum1的值的。题目转变为求两数之和

对特殊情况优化后，执行用时直接减小到了5ms，击败了99.75%的用户， 之前没有优化的时候是31ms,只击败了81.64%的用户，内存消耗都差不多98.8%，用了37MB左右。

private static List<List<Integer>> solve1(int[] nums, int target){
        List<List<Integer>> result = new ArrayList<>();
        //数组排序
        Arrays.sort(nums);
        int len = nums.length;
        for (int i=0; i<len-3; i++){   //小于len-3是因为最i最大也是len-4，后面才会有三个值
            if (i>0 && nums[i]==nums[i-1])       //跳过相同的nums[i]。
                continue;
            //网上看到的针对特殊情况的优化。
            //若是i开始连续的四个数都大于target,那肯定没有满足情况的数，直接跳出。
            if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target)
                break;
            //若是i和最大的三个数相加都比target小，那说明i太小了，直接跳过这一个值往下走
            if(nums[i]+nums[len-1]+nums[len-2]+nums[len-3]<target)
                continue;
            for (int j=i+1; j<len-2; j++){
                if (nums[j]==nums[j-1] && j-i>1 )       //跳过相同的nums[j]。此处需要注意的是第一个j的判断：j-i>1
                    continue;
                //同理，此处也可以优化
                if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)
                    break;
                //若是i和最大的两个数相加都比target小，那说明j太小了，直接跳过这一个值往下走
                if(nums[i]+nums[j]+nums[len-1]+nums[len-2]<target)
                    continue;
                int leave = target-nums[i]-nums[j];
                //从【j+1,len-1】寻找剩下的两个数
                int left = j+1, right = len-1;
                while (left<right){
                    if (nums[left]+nums[right]==leave){
                        result.add(new ArrayList<>(Arrays.asList(nums[i],nums[j],nums[left],nums[right])));
                        while (left<right && nums[left+1]==nums[left]){
                            left++;                   //跳过相同的nums[left]。
                        }
                        while (left<right && nums[right-1]==nums[right]){   //跳过相同的nums[right]。
                            right--;
                        }
                        left++;
                        right--;
                    }else if(nums[left]+nums[right]<leave){
                        left++;
                    }else {
                        right--;
                    }

                }
            }
        }

        return result;
    }