算法：四数之和

最新推荐文章于 2024-02-05 15:12:00 发布

盼盼编程

最新推荐文章于 2024-02-05 15:12:00 发布

阅读量389

点赞数 1

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/ma2595162349/article/details/111263677

版权

数据结构与算法专栏收录该内容

196 篇文章 22 订阅

订阅专栏

该博客介绍了如何通过排序和双指针算法解决LeetCode上的四数之和问题。给定一个整数数组和一个目标值，我们需要找到所有使四数之和等于目标值的不重复四元组。博主提供了详细的代码实现，利用有序数组的特性进行剪枝，提高了算法效率。

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

题目：

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。注意：答案中不可以包含重复的四元组。示例：给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。满足要求的四元组集合为： [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]

//四数之和

func fourSum(nums []int, target int) (quadruplets [][]int) {
    //对数组进行排序,注意有序数组这个规律进行剪枝
    sort.Ints(nums)
    n := len(nums)

    //遍历整个数组
    for i := 0; i < n-3 && nums[i]+nums[i+1]+nums[i+2]+nums[i+3] <= target; i++ {
        //由于是有序数组，这里进行剪枝
        if i > 0  &&  nums[i]  ==  nums[i-1]  ||  nums[i]+nums[n-3]+nums[n-2]+nums[n-1] < target {
            continue
        }

        for j := i + 1; j < n-2 && nums[i]+nums[j]+nums[j+1]+nums[j+2] <= target; j++ {
            //由于是有序数组这里进行剪枝
            if j > i+1 && nums[j] == nums[j-1] || nums[i]+nums[j]+nums[n-2]+nums[n-1] < target {
                continue
            }
       
            //双指针
            for left, right := j+1, n-1; left < right; {
                if sum := nums[i] + nums[j] + nums[left] + nums[right]; sum == target {
                    //得到结果  
                    quadruplets = append(quadruplets, []int{nums[i], nums[j], nums[left], nums[right]})

                    for left++; left < right && nums[left] == nums[left-1]; left++ {
                    }

                    for right--; left < right && nums[right] == nums[right+1]; right-- {
                    }
                } else if sum < target {
                    left++
                } else {
                    right--
                }
            }

        }

    }

    return
}

链接：https://leetcode-cn.com/problems/4sum/solution/si-shu-zhi-he-by-leetcode-solution/