【leetcode18】四数之和--go语言实现

最新推荐文章于 2023-07-04 23:15:00 发布

胡桃木子

最新推荐文章于 2023-07-04 23:15:00 发布

阅读量271

点赞数

分类专栏： leetcode 文章标签： leetcode 算法贪心算法

本文链接：https://blog.csdn.net/weixin_42918559/article/details/125113367

版权

leetcode 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

该博客主要介绍了如何利用双指针法解决寻找数组中和为目标值的不重复四元组的问题。在15.三数之和的基础上，通过增加一层循环，有效地将时间复杂度降低到O(n^3)。代码示例展示了如何处理数组，避免重复，并在找到符合条件的四元组时将其添加到结果集中。这种方法特别适用于处理四数之和的搜索问题。

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

题目描述：

给你一个由 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]]

思路实现：

四数之和，和15.三数之和 (opens new window)是一个思路，都是使用双指针法,
基本解法就是在15.三数之和 (opens new window) 的基础上再套一层for循环。对于15.三数之和 (opens new window)双指针法就是将原本暴力 $O(n^3)$ 的解法，降为 $O(n^2)$ 的解法，四数之和的双指针解法就是将原本暴力 $O(n^4)$ 的解法，降为 $O(n^3)$ 的解法。

代码实现：

func fourSum(nums []int, target int) [][]int {
	res:=[][]int{}
	lens:=len(nums)
	if lens<=3{
		return res
	}
	sort.Ints(nums)
	if target>0&&nums[0]>=target{//如果第一个大于target，不满足,
		return res
	}
	if target<0&&nums[lens-1]<=target{//如果第一个大于target，不满足
		return res
	}
	for i:=0;i<lens-2;i++{
		if target>0&&nums[i]>=target{//退出条件
			break
		}
		if i>0&&nums[i]==nums[i-1]{//去重处理
			continue
		}
		for j:=i+1;j<lens-1;j++{
			
			if j!=i+1&&nums[j]==nums[j-1]{//去重处理
				continue
			}
			k,l:=j+1,lens-1
			for k<l{
				if k!=j+1&&nums[k]==nums[k-1]{
                    k++
					continue
				}
				if l!=lens-1&&nums[l]==nums[l+1]{
                    l--
					continue
				}
				if nums[i]+nums[j]+nums[k]+nums[l]<target{
					k++
				}else if nums[i]+nums[j]+nums[k]+nums[l]>target{
					l--
				}else {
					temp:=[]int{nums[i],nums[j],nums[k],nums[l]}
					res=append(res,temp)
					k++
					l--
				}
			}
		}
	}
	return res
}