求解四数之和

最新推荐文章于 2024-08-23 01:05:03 发布

海边画家

最新推荐文章于 2024-08-23 01:05:03 发布

阅读量96

点赞数

分类专栏： 100天复习笔记文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_44050588/article/details/129208126

版权

100天复习笔记专栏收录该内容

14 篇文章 0 订阅

订阅专栏

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

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
       
        for (int i = 0; i < nums.length; i++) {
		
            // nums[i] > target 直接返回, 剪枝操作
            if (nums[i] > 0 && nums[i] > target) {
                return result;
            }
		
            if (i > 0 && nums[i - 1] == nums[i]) {    // 对nums[i]去重
                continue;
            }
            
            for (int j = i + 1; j < nums.length; j++) {

                if (j > i + 1 && nums[j - 1] == nums[j]) {  // 对nums[j]去重
                    continue;
                }

                int left = j + 1;
                int right = nums.length - 1;
                while (right > left) {
		    // nums[k] + nums[i] + nums[left] + nums[right] > target int会溢出
                    long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum > target) {
                        right--;
                    } else if (sum < target) {
                        left++;
                    } else {
                        result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        // 对nums[left]和nums[right]去重
                        while (right > left && nums[right] == nums[right - 1]) right--;
                        while (right > left && nums[left] == nums[left + 1]) left++;

                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
}

利用双指针法解决四数相加之和，并让元组不重复，首先对数组进行排序，关键点在于跳过重复的值，举例来说1 2 2 2 5 5 6，若target11 为第一次循环，那么i指向1 j指向2 left指向2 right指向6，此时nums[i]+nums[j]+nums[left]+nums[right]=11,那么此时nums[left] = nums[left+1]，最后结果会跑到第三个2，然后left++，故下次left会从5开始，依次往复即可。