leetcode算法题-数组-四数之和

题目描述

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。
注意：答案中不可以包含重复的四元组。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目思路

• 总的来说，本题的思想和三数之和的思想是一样的
• 详细思路看我的三数之和的博客 点我

代码实现

package com.leetcode.数组;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * Author:markusZhang
 * VM Args:
 * Date:Create in 2020/2/2 13:03
 */
public class 四数之和 {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> aList = new ArrayList<>();
        if(nums==null || nums.length<4){
            return aList;
        }
        Arrays.sort(nums);
        for(int i=0;i<nums.length-3;i++){
            if(i>0 && nums[i]==nums[i-1]){
                continue;
            }
            int min = nums[i]+nums[i+1]+nums[i+2]+nums[i+3];
            if(min>target){
                continue;
            }
            int max = nums[i]+nums[nums.length-1]+nums[nums.length-2]+nums[nums.length-3];
            if(max<target){
                continue;
            }
            for(int k=i+1;k<nums.length-2;k++){
                if(k>i+1 && nums[k]==nums[k-1]){
                    continue;
                }
                int h = nums.length-1;
                int l = k+1;
                int min1 = nums[i]+nums[k]+nums[k+1]+nums[k+2];
                if(min1>target){
                    continue;
                }
                int max1 = nums[i]+nums[k]+nums[h-1]+nums[h];
                if(max1<target){
                    continue;
                }
                while(l<h){
                    int sum = nums[i]+nums[k]+nums[l]+nums[h];
                    if(sum==target){
                        aList.add(Arrays.asList(nums[i],nums[k],nums[l],nums[h]));
                        l++;
                        while(l<h && nums[l]==nums[l-1]){
                            l++;
                        }
                        h--;
                        while(l<h && nums[h]==nums[h+1]){
                            h--;
                        }
                    }else if(sum<target){
                        l++;
                    }else{
                        h--;
                    }
                }
            }
        }
        return aList;
    }

    public static void main(String[] args) {
        int nums[] = {1, 0, -1, 0, -2, 2};
        System.out.println(new 四数之和().fourSum(nums,0));
    }
}