LeetCode18 四数之和

LeetCode18 四数之和题解（附伪代码步骤）

题目

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

解题思路

前面写过了LeetCode第1题“两数之和”以及LeetCode第15题“三数之和”，这道题就比较简单了，思路与“三数之和”那道题思路类似（可以先看一下那道题的思路: LeetCode15 三数之和题解）
首先对原数组进行排序，用这种方法防止结果出现重复的四元组。之后遍历nums数组，在遍历到第i个元素nums[i]时，将第i+1的元素到最后一个元素组成一个新数组，在这个新数组中寻找三个数使得满足他们的和等于target - nums[i]，之后的步骤就可以调用三数之和的函数进行寻找。
不过需要注意的是，为了防止出现重复的四元组，在遍历nums数组时，如果nums[i]==nums[i-1]时，需要跳过此次遍历。

伪代码步骤

step1：对nums数组进行排序
step2：遍历nums数组：
step3：如果当前数组值等于前一个数组值，那么跳过此次循环
step4：将当前数组值之后的数组值组成一个新的数组
step5：调用threesum()函数进行检索是否有满足条件的三元组
step6：将找到的三元组分别与当前值组成四元组

代码展示

public static List<List<Integer>> Twosum(int start,int end,int nums[],int value,int target)
    {
        List<List<Integer>> list = new ArrayList<>();
        while(start<end)
        {
            if(start<end)
            {
                int a = nums[start];
                int b = nums[end];
                while(nums[start]+nums[end]>(target)&&start<end)
                {
                    end--;
                }
                while(nums[start]+nums[end]<(target)&&start<end)
                {
                    start++;
                }
                if(nums[start]+nums[end]==(target)&&start<end)
                {
                    List<Integer> alist = new ArrayList<>();
                    alist.add(value);
                    alist.add(nums[start]);
                    alist.add(nums[end]);
                    list.add(alist);
                    //防止重复
                    while(start!= nums.length-1&&nums[start+1]==nums[start])
                        start++;
                    while(end!=0&&nums[end]==nums[end-1])
                        end--;
                    if(start>=end)
                        break;
                    start++;
                    end--;
                }
            }

        }
        return list;
    }



    public static List<List<Integer>> threeSum(int[] nums,int target) {
        List<List<Integer>> list = new ArrayList<>();
        Arrays.sort(nums);

        for(int i=0;i< nums.length;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            List<List<Integer>> alist = Twosum(i+1, nums.length-1,nums,nums[i],target - nums[i]);
            for (int j=0;j<alist.size();j++)
                list.add(alist.get(j));

        }
        return list;
    }

    public static List<List<Integer>> fourSum(int[] nums,int target)
    {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);

        for (int i=0;i< nums.length-2;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            int a[] = new int[nums.length-i-1];
            for (int j = 0;j<a.length;j++)
            {
                a[j] = nums[i+j+1];
            }
            List<List<Integer>> alist = threeSum(a,target-nums[i]);
            for (int j=0;j<alist.size();j++)
            {
                List<Integer> list = new ArrayList<>();
                list.add(nums[i]);
                for (int t=0;t<alist.get(j).size();t++)
                {
                    list.add(alist.get(j).get(t));
                }
                res.add(list);
            }
        }
        return res;
    }

如果觉得这篇题解有用的话就给我点个赞吧，有什么也可以在评论区交流交流~