写点什么

[Day46]-[数组]- 三数之和

作者:方勇(gopher)
  • 2022 年 5 月 28 日
  • 本文字数:766 字

    阅读完需:约 3 分钟

15. 三数之和

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

 

示例 1:

输入:nums = [-1,0,1,2,-1,-4]输出:[[-1,-1,2],[-1,0,1]]
复制代码

示例 2:

输入:nums = []输出:[]
复制代码

示例 3:

输入:nums = [0]输出:[]
复制代码

题解:


// 时间复杂度 O(N²) 空间复杂度O(logN)用于排序func TestThreeSum(t *testing.T) {	var nums = []int{-1, 0, 1, 2, -1, -4} // 目标数组	var ans [][]int                       // 结果集	sort.Ints(nums)                       // 先要排序,防止重复结果 排序用的Golang的快排	n := len(nums)                        // 数组长度	for first := 0; first < n; first++ {  // 第一重循环		if first > 0 && nums[first] == nums[first-1] { // 排序后数据 如果存在重复的 右移指针			continue		}		target := -1 * nums[first]                      // 和为0,所以取相反数		third := n - 1                                  // 第三个指针		for second := first + 1; second < n; second++ { // 第二重循环 采用双指针 第二个指针右移,第三个指针左移			if second > first+1 && nums[second] == nums[second-1] { // 排序后的数据,如果存在重复,右移指针				continue			}			for second < third && nums[second]+nums[third] > target { // 由于是排好序的 所以如果第三个数过大 就左移指针				third--			}			if second == third { // 如果指针重合 说明没有找到跳出当前循环				break			}			if nums[second]+nums[third] == target { // 找到了结果放到结果集				ans = append(ans, []int{nums[first], nums[second], nums[third]})			}		}	}	t.Log(ans)}
复制代码


用户头像

Dead or Alive. 生存战斗是知识的源泉! 2018.11.08 加入

我是一名SRE哨兵,目前是好大夫基础架构部高级工程师。专注于 SRE,微服务、中间件的稳定性和可用性建设,整体负责好大夫服务治理云平台的设计和搭建!

评论

发布
暂无评论
[Day46]-[数组]-三数之和_数组_方勇(gopher)_InfoQ写作社区