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

【leetcode18】四数之和--go语言实现,第1张

题目描述:

给你一个由 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^3) O(n3)的解法,降为 O ( n 2 ) O(n^2) O(n2)的解法,四数之和的双指针解法就是将原本暴力 O ( n 4 ) O(n^4) O(n4)的解法,降为 O ( n 3 ) O(n^3) O(n3)的解法。

代码实现:
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
}

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/1324258.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-06-12
下一篇 2022-06-12

发表评论

登录后才能评论

评论列表(0条)

保存