[LeetCode-Golang] 15. 三数之和
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
排序 + 双指针
1.nums从小到大排序,i,j,k分别代表元素a,b,c的索引
2.先固定i,i的范围是区间[0, len(nums)-2),注意要对a去重,如果nums[i] == nums[i-1]将可能出现三元组重复
3.固定i后,取j = i+1,k =len(nums)-1,j和k在j<k的情况下(j>i和j<k已经保证j和k在移动过程中不会数组越界)作为双指针进行遍历,判断a+b+c
3.1如果a+b+c==0,则添加三元组,然后j++、k--,并且注意对b,c去重,原因同2中的a
3.2如果a+b+c>0,则k--
3.3如果a+b+c<0,则j++
4.注意题目没有说明nums范围,因此需要特殊处理nums == nil 和len(nums)<3,返回nil或者[][]int{}都可以
复杂度分析
时间复杂度:O(n2)
空间复杂度:不算返回的三元组为O(1)
代码
func threeSum(nums []int) [][]int { if nums == nil || len(nums) < 3 { return nil } // 从小到大排序 sort.Ints(nums) ans := [][]int{} for i := 0; i < len(nums)-2; i++ { // a只能是<=0,不能>0。当a>0,由于非递减排序,直接退出循环 if nums[i] > 0 { break } // 给a去重 if i > 0 && nums[i] == nums[i-1] { continue } // 固定i,移动j和k j := i+1 k := len(nums)-1 for j < k { // 判断和与0的关系 sum := nums[i]+nums[j]+nums[k] if sum == 0 { // 追加三元组 ans = append(ans, []int{nums[i], nums[j], nums[k]}) // j++, k--并且对j和k去重 for j < k && nums[j] == nums[j+1] { j++ } for j <k && nums[k] == nums[k-1] { k-- } j++ k-- } else if sum > 0 { // k向左移动 k-- } else { j++ } } } return ans }
执行结果
执行用时 :36 ms, 在所有 Go 提交中击败了79.60% 的用户
内存消耗 :6.9 MB, 在所有 Go 提交中击败了52.13%的用户