18. 四数之和

算法题（程序员面试宝典）

解题思路主要来源于leetcode官方与《程序员面试宝典》&labuladong

18. 四数之和

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

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

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2：

输入：nums = [], target = 0
输出：[]

提示：

0 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

解题方法

解题思路1

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        //首先对数组进行排序操作（升序）
        Arrays.sort(nums);
        return nSumTarget(nums,4,0,target);
    }

    // nums:数组  n：n个元素之和  数组的遍历起始位置：start  target:目标值  
    public List<List<Integer>> nSumTarget(int[] nums,int n,int start,int target){
        List<List<Integer>> res = new ArrayList<>();
        int len = nums.length;
        //合法校验   n<2 或者 数组的长度小于n
        if(n<2||len<n)
            return res;

        int left=start;
        int right=len-1;
        if(n==2){
            //二数之和
            while(left<right){
                int sum = nums[left]+nums[right];
                if(sum==target){
                    List<Integer> val = new ArrayList<>();
                    val.add(nums[left]);
                    val.add(nums[right]);
                    res.add(val);
                    //去除重复结果操作
                    //与nums[left]相同，则跳过
                    while(left<right&&nums[left]==nums[left+1])
                        left +=1;
                    //与nums[right]相同，则跳过
                    while(left<right&&nums[right]==nums[right-1])
                        right -=1;
                    left = left+1;
                    right = right-1;
                }else if(sum<target){
                    //增加left
                    while(left<right&&nums[left]==nums[left+1])
                        left +=1;
                    left +=1;
                }else if(sum>target){
                    while(left<right&&nums[right]==nums[right-1])
                        right -=1;
                    right = right-1;
                }
            }
        }else if(n>2){
            // n 数之和
            for(int i=start;i<nums.length;i++){

                List<List<Integer>> cur = nSumTarget(nums,n-1,i+1,target-nums[i]);
                for(List<Integer> item:cur){
                    item.add(nums[i]);
                    res.add(item);
                }
                //去除重复的结果
                while(i<(nums.length-1)&&nums[i]==nums[i+1])
                    i++;
            }
        }

        return res;
    }
}