求和为给定值的四元组

这篇博客主要探讨了如何使用Java实现四数之和的问题,针对给定整数数组找到所有使得它们的和等于给定目标值的四元组。作者首先介绍了两种方法,一种是基于排序和双指针的递归解决方案,另一种是改进后的双层循环加双指针方法,特别考虑了重复元素的情况。博客还包含了一个测试用例,用于验证算法的正确性。
摘要由CSDN通过智能技术生成
package main.java.demo18;

import org.junit.Test;

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

public class FourSum {


    private List<List<Integer>> result = new ArrayList<>();
    private Boolean[][] booleans = new Boolean[201][201];

    /**
     * 解题思路: 先排序, 在双指针, 两层双指针, 这样写会出现一些问题, 比如输入{2,2,2,2,2} target = 8.
     *          现在脑子有点混乱, 之后刷到类似题再回来解决, 下面方法使用简单的两层循环加双指针.
     *
     * @param nums
     * @param target
     * @return
     */
    public List<List<Integer>> fourSum_1(int[] nums, int target) {

        int n = nums.length;
        Arrays.sort(nums);
        for (int i = 0; i < 201; i++) {
            for (int j = 0; j < 201; j++) {
                booleans[i][j] = false;
            }
        }

        recursion(0, n - 1, nums, target);

        return result;
    }

    public void recursion(int outerLeft, int outerRight, int[] nums, int target) {
        if (outerLeft >= outerRight) {
            return;
        }
        int n = nums.length;
        if (outerLeft > 0 && nums[outerLeft - 1] == nums[outerLeft]) {
            if (booleans[outerLeft + 1][outerRight] == true) {
                return;
            }
            recursion(outerLeft + 1, outerRight, nums, target);
            booleans[outerLeft + 1][outerRight] = true;
        }
        if (outerRight < n - 2 && nums[outerRight + 1] == nums[outerRight]) {
            if (booleans[outerLeft][outerRight - 1] == true) {
                return;
            }
            recursion(outerLeft, outerRight - 1, nums, target);
            booleans[outerLeft][outerRight - 1] = true;
        }
        int innerSum = target - nums[outerLeft] - nums[outerRight];
        int innerLeft = outerLeft + 1;
        int innerRight = outerRight - 1;
        while (innerLeft < innerRight) {
            if (innerLeft > outerLeft + 1 && nums[innerLeft - 1] == nums[innerLeft]) {
                innerLeft++;
                continue;
            }
            if (innerRight < outerRight - 1 && nums[innerRight + 1] == nums[innerRight]) {
                innerRight--;
                continue;
            }
            int val = innerSum - nums[innerLeft] - nums[innerRight];
            if (val == 0) {
                List<Integer> current = new ArrayList<>();
                current.add(nums[outerLeft]);
                current.add(nums[outerRight]);
                current.add(nums[innerLeft]);
                current.add(nums[innerRight]);
                result.add(current);
                innerLeft++;
                innerRight--;
            } else if (val > 0) {
                innerLeft++;
            } else {
                innerRight--;
            }
        }
        // -2,-1,0,0,1,2
        if (booleans[outerLeft + 1][outerRight] == false) {
            recursion(outerLeft + 1, outerRight, nums, target);
            booleans[outerLeft + 1][outerRight] = true;
        }
//        while (outerLeft > 0 && nums[outerLeft - 1] == nums[outerLeft]) {
//            outerLeft++;
//        }
        if (booleans[outerLeft][outerRight - 1] == false) {
            recursion(outerLeft, outerRight - 1, nums, target);
            booleans[outerLeft][outerRight - 1] = true;
        }
    }

    public List<List<Integer>> fourSum_2(int[] nums, int target) {

        int n = nums.length;
        int sum;
        int leftPoint;
        int rightPoint;
        Arrays.sort(nums);


        for (int i = 0; i < n - 3; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for (int j = i + 1; j < n - 2; j++) {
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }
                sum = target - nums[i] - nums[j];
                leftPoint = j + 1;
                rightPoint = n - 1;
                while (leftPoint < rightPoint) {
                    if (leftPoint > j + 1 && nums[leftPoint] == nums[leftPoint - 1]) {
                        leftPoint++;
                        continue;
                    }
                    if (rightPoint < n - 1 && nums[rightPoint] == nums[rightPoint + 1]) {
                        rightPoint--;
                        continue;
                    }
                    int val = sum - nums[leftPoint] - nums[rightPoint];
                    if (val == 0) {
                        List<Integer> current = new ArrayList<>();
                        current.add(nums[i]);
                        current.add(nums[j]);
                        current.add(nums[leftPoint]);
                        current.add(nums[rightPoint]);
                        result.add(current);
                        rightPoint--;
                        leftPoint++;
                    } else if (val < 0) {
                        rightPoint--;
                    } else {
                        leftPoint++;
                    }
                }
            }
        }

        return result;

    }



        @Test
    public void formSumTest() {
//        System.out.println(fourSum_1(new int[]{-3,-1,0,2,4,5}, 0));
//        System.out.println(fourSum_2(new int[]{-2,-1,0,0,1,2}, 0));
        System.out.println(fourSum_2(new int[]{2,2,2,2,2}, 8));
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值