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> fourSum(int[] nums, int target) {
     List> 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 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> fourSum(int[] nums, int target) {
        List> result = new ArrayList<>();
        if(nums.length<4){
            return result;
        }
        Arrays.sort(nums);
        for(int i=0;i 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 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

你可能感兴趣的:(LeetCode笔记)