LeetCode 18.四数之和

问题描述

给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[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 = [2,2,2,2,2], target = 8

输出:[[2,2,2,2]]

Leetcode链接

解题思路

  1. 排序:首先对数组进行排序,这样可以方便地使用双指针来查找满足条件的四元组。

  2. 遍历:使用两层循环来固定前两个数,然后在剩下的数组中使用双指针来查找另外两个数。

  3. 去重:为了避免重复的四元组,需要在每次遍历时跳过相同的数。

code

java 复制代码
/**
 * @package: _03hash._18fourSum
 * @description:18.四数之和
 * @author: Yunyang
 * @date: 2024/10/23  14:10
 * @version:1.0
 **/
public class TestFourSum {
    public static void main(String[] args) {
        Solution solution = new Solution();

        int[] nums1 = {1, 0, -1, 0, -2, 2};
        int target1 = 0;
        System.out.println(solution.fourSum(nums1, target1)); // 输出: [[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

        int[] nums2 = {2, 2, 2, 2, 2};
        int target2 = 8;
        System.out.println(solution.fourSum(nums2, target2)); // 输出: [[2,2,2,2]]

        int[] nums3 = {1000000000,1000000000,1000000000,1000000000};
        int target3= -294967296;
        System.out.println(solution.fourSum(nums3, target3));
    }
}

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        // 初始化result数组
        List<List<Integer>> result = new ArrayList<>();

        if(nums == null || nums.length < 4){
            return result;
        }

        // 对数组 nums 进行排序
        Arrays.sort(nums);

        // 使用两层循环来固定前两个数 nums[i] 和 nums[j],
        // 然后在剩下的数组中使用双指针 left 和 right 来查找另外两个数
        int n = nums.length;
        for (int i = 0; i < n; i++) {
            // 提前终止条件
            if (nums[i] > target && nums[i] >= 0){
               break;
            }

            // 跳过重复的元素
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }

            for (int j = i + 1; j < n - 1; j++) {
                // 提前终止条件
                if(nums[i] + nums[i] > target && nums[i] + nums[j] >= 0){
                    break;
                }

                // 跳过重复的元素
                if(j > i + 1 && nums[j] == nums[j - 1]){
                    continue;
                }

                int left = j + 1;
                int right = n - 1;

                while (left < right){
                    long sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum == Integer.MIN_VALUE){
                        return result;
                    }

                    if(sum > target){
                        right--;
                    } else if(sum < target){
                        left++;
                    } else {
                        result.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));

                        // 跳过重复的元素
                        while (left < right && nums[left] == nums[left + 1]){
                            left++;
                        }

                        // 跳过重复的元素
                        while (left < right && nums[right] == nums[right - 1]){
                            right--;
                        }
                        left++;
                        right--;
                    }
                }
            }

        }
        return result;
    }
}

代码解释

  1. 整数溢出处理 :在计算 sum 时,使用 long 类型来避免整数溢出。
  2. 提前终止条件 :在 ij 的循环中,添加提前终止条件,以减少不必要的计算。
  3. 去重:在每次遍历时跳过重复的元素,避免生成重复的四元组。
相关推荐
single59420 分钟前
【综合算法学习】(第十篇)
java·数据结构·c++·vscode·学习·算法·leetcode
DC妙妙屋27 分钟前
10.24.2024刷华为OD C题型(四) -- 对象list按照多个属性排序
1024程序员节
bitenum1 小时前
qsort函数的学习与使用
c语言·开发语言·学习·算法·visualstudio·1024程序员节
字节卷动1 小时前
【牛客算法】某司面试算法题:循环右移二叉树
数据结构·算法·leetcode·面试·牛客
pengpeng022 小时前
力扣每日一题 3211. 生成不含相邻零的二进制字符串
算法·leetcode
EmotionFlying2 小时前
(11)(2.1.6) Hobbywing DroneCAN ESC(一)
copter·ardupilot·1024程序员节·电调和电机
伏飞而行2 小时前
六、元素应用CSS的习题
1024程序员节
idealzouhu2 小时前
Spring Boot 实现文件上传下载功能
java·spring boot·1024程序员节
sudo_Ene3 小时前
Isaac Sim Docker 部署并使用过程记录
笔记·学习·docker·1024程序员节·isaacsim