18. 四数之和

给你一个n 个整数组成的数组 nums ,和一个目标值 target 。请你找出返回满足下述全部条件重复四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一对应,则认为两个四元组重复):

可以任意顺序 返回答案

示例 1:

输入nums = [1,0,-1,0,-2,2], target = 0

输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2:

输入nums = [2,2,2,2,2], target = 8

输出:[[2,2,2,2]]

算法思路

  1. 排序
  2. 依次固定一个a
  3. a后面的区间内, 利用“三数之和”找到三个数, 是这三个数的和等于target - a即可(三数之和题解点击这里)

处理细节问题

处理重复四元组:

Java代码

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> lists = new ArrayList<>();
        int n = nums.length;
        int i = 0;
        while(i < n - 3) {
            if(nums[i] > 0 &amp;&amp; nums[i] > target) {
                break;
            }
            //固定数a
            int a = nums[i];
            long targetA = target - a;
            int j = i + 1;
            while (j < n - 2) {
                if(nums[j] > 0 &amp;&amp; nums[j] > targetA) {
                    break;
                }
                //固定数b
                int b = nums[j];
                long targetB = targetA - b;
                int left = j + 1;
                int right = n - 1;
                while (left < right) {
                    int sum = nums[left] + nums[right];
                    if(sum < targetB) {
                        left++;
                    } else if (sum > targetB) {
                        right--;
                    } else {
                        lists.add(Arrays.asList(a, b, nums[left], nums[right]));
                        left++;
                        right--;
                        //去重, 不能越界
                        while (left < right && nums[left] == nums[left - 1]) {
                            left++;
                        }
                        while (left < right && nums[right] == nums[right + 1]) {
                            right--;
                        }
                    }
                }
                //去重, 不能越界
                j++;
                while (j < n - 2 && nums[j] == nums[j - 1]) {
                    j++;
                }
            }
            //去重, 不能越界
            i++;
            while (i < n - 3 && nums[i] == nums[i - 1]) {
                i++;
            }
        }
        return lists;
    }
}

时间复杂度: O(N3) 空间复杂度: O(log2N)

原文地址:https://blog.csdn.net/m0_73594607/article/details/134746018

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任

如若转载,请注明出处:http://www.7code.cn/show_35804.html

如若内容造成侵权/违法违规/事实不符,请联系代码007邮箱suwngjj01@126.com进行投诉反馈,一经查实,立即删除

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注