leetcode 18. 四數之和
阿新 • • 發佈:2022-01-07
一、題目
給你一個由 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]]
二、解法
本題思路與三數之和一樣,用雙指標法,將暴力求解的\(O(n^4)\)降為\(O(n^3)\)
但本題不是求0,而是求target,target可能為負數,所以不能剪枝。
func fourSum(nums []int, target int) [][]int { len:=len(nums) if len<4{ return nil } sort.Ints(nums) ans:=make([][]int,0) for i:=0;i<len;i++{ if i>0&&nums[i]==nums[i-1]{ continue } for j:=i+1;j<len;j++{ if j>i+1&&nums[j]==nums[j-1]{ continue } k,l:=j+1,len-1 for k<l{ sum:=nums[i]+nums[j]+nums[k]+nums[l] if sum==target{ ans=append(ans,[]int{nums[i],nums[j],nums[k],nums[l]}) k++ l-- for k<l&&nums[k]==nums[k-1]{ k++ } for k<l&&nums[l]==nums[l+1]{ l-- } }else if sum>target{ l-- }else{ k++ } } } } return ans }