给定一个包含 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) [][]int {
var result [][]int
sort.Ints(nums)
for i := 0; i < len(nums); i++ {
for j := len(nums) - 1; j > i+2; j-- {
m, n := i+1, j-1
for m < n {
sum := nums[i] + nums[j] + nums[m] + nums[n]
if sum == target {
numArr := []int{nums[i], nums[m], nums[n], nums[j]}
flag := true
for _, v := range result {
if v[0] == numArr[0] && v[1] == numArr[1] && v[2] == numArr[2] && v[3] == numArr[3] {
flag = false
}
}
if flag {
result = append(result, numArr)
}
m, n = m+1, n-1
} else if sum < target {
m++
} else {
n--
}
}
}
}
return result
}