n数之和-三数之和，四数之和

浮尘~

于 2022-09-09 00:26:48 发布

阅读量140

点赞数

分类专栏：题解文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/qq_39367862/article/details/126774791

版权

题解专栏收录该内容

13 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用Java编程解决找到数组中四个数的组合，其和等于给定目标值的问题。代码首先将输入的整数数组转换为整型数组并进行排序，然后通过递归方式查找所有可能的四数组合。在查找过程中，采用了类似于快速排序的去重策略，确保了结果的唯一性。这种方法适用于大数据集，能有效减少计算量。

摘要由CSDN通过智能技术生成

必须先排序


import java.util.*;

public class Solution {
    public ArrayList<ArrayList<Integer>> fournumber(ArrayList<Integer> nums, int target) {
        int[] num = nums.stream().mapToInt(Integer::intValue).toArray();
        Arrays.sort(num);
        return nsum(num, 4, 0, target);

    }

    public ArrayList<ArrayList<Integer>> nsum(int[] num, int n, int start, int target) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        int len = num.length;
        if (len < n) return res;
        if (n == 2) {
            int left = start;
            int right = len - 1;
            while (left < right) {
                int sum = num[left] + num[right];
                if (sum == target) {
                    ArrayList<Integer> temp = new ArrayList<>();
                    temp.add(num[left]);
                    temp.add(num[right]);
                    res.add(temp);
                    //类似快排，去重
                    while (left < right && num[left] == num[left + 1]) left++;
                    while (left < right && num[right] == num[right - 1]) right--;
                    left++;
                    right--;
                } else if (sum < target) {
                    left++;
                } else {
                    right--;
                }
            }
        } else {
            for (int i = start; i < len - n + 1; i++) {
            //去重
                if (i > start && num[i] == num[i - 1]) continue;
                ArrayList<ArrayList<Integer>> sub = nsum(num, n - 1, i + 1, target - num[i]);
                for (ArrayList<Integer> arr : sub) {
                    arr.add(num[i]);
                    res.add(arr);
                }
            }
        }
        return res;
    }
}