算法:四数之和
題目:?
給定一個包含 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/
《新程序員》:云原生和全面數字化實踐50位技術專家共同創作,文字、視頻、音頻交互閱讀總結