LeetCode笔记——18四数之和

题目：

给定一个包含 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]
]

思路：以下是网上大神的代码。首先对数组进行排序，先确定第一个数和最后一个数，之后确定中间两个数。在整个过程中要确保不会有相同的数重复出现。

代码：

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
     List<List<Integer>> result = new LinkedList<>();
        if (nums == null || nums.length < 4) {
            return result;
        }

        Arrays.sort(nums); // 对数组进行排序

        for (int i = 0; i < nums.length - 3; i++) { // 第一个加数
            if ( i > 0 && nums[i] == nums[i - 1]) { // 第一个加数使用不重复
                continue;
            }

            for (int j = nums.length - 1; j > i + 2 ; j--) { // 第四个加数
                if (j < nums.length - 1 && nums[j] == nums[j + 1]) { // 第四个加数使用不重复
                    continue;
                }

                int start = i + 1; // 第二个加数
                int end = j - 1; // 第三个加数
                int n = target - nums[i] - nums[j];

                while (start < end) {
                    if (nums[start] + nums[end] == n) {
                        List<Integer> four = new ArrayList<>(4);
                        four.add(nums[i]);
                        four.add(nums[start]);
                        four.add(nums[end]);
                        four.add(nums[j]);

                        result.add(four);

                        do {
                            start++;
                        } while (start< end && nums[start] == nums[start - 1]); // 保证再次使用第二个数不重复

                        do {
                            end--;
                        } while (start < end && nums[end] == nums[end + 1]); // 保证再次使用第三个数不重复
                    } else if (nums[start] + nums[end] < n) {
                        do {
                            start++;
                        } while (start< end && nums[start] == nums[start - 1]); // 保证再次使用第二个数不重复
                    } else {
                        do {
                            end--;
                        } while (start < end && nums[end] == nums[end + 1]); // 保证再次使用第三个数不重复
                    }
                }
            }
        }

        return result;   
    }
}

执行时间最短的用例：

基本思路是相同的，只不过跳过了几个特殊的情况。

lass Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> result = new ArrayList<>();
        if(nums.length<4){
            return result;
        }
        Arrays.sort(nums);
        for(int i=0;i<nums.length-3;i++){
            if(nums[i] + nums[i+1] + nums[i+2] + nums[i+3] > target){ //nums[i] is too large
                break;
            }
            if(nums[i] + nums[nums.length-1] + nums[nums.length-2] + nums[nums.length-3] < target){//nums[i] is too small
                continue;
            }
            if(i>0 && nums[i]==nums[i-1]){  //跳过重复项
                continue;
            }
            for(int j=i+1;j<nums.length-2;j++){
                if(nums[i]+nums[j]+nums[j+1]+nums[j+2] > target){   //nums[j] is too large
                    break;
                }
                if(nums[i]+nums[j]+nums[nums.length-1] + nums[nums.length-2] < target){ // nums[j] is too small
                    continue;
                }
                if(j>i+1 && nums[j]==nums[j-1]){    //跳过重复项
                    continue;
                }
                int left = j+1;
                int right = nums.length-1;
                while(left<right){
                    int sum = nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum==target){
                        result.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));
                        while(left<right && nums[left]==nums[left+1]){
                            left++;
                        }
                        while(left<right && nums[right]==nums[right-1]){
                            right--;
                        }
                        left++;
                        right--;
                    }else if(sum<target){
                        left++;
                    }else{
                        right--;
                    }
                }
            }
        }
        return result;
    }
}