[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%的用户

原文地址:https://www.cnblogs.com/nini-skyrim/p/12501529.html

时间: 2024-11-09 00:57:28

[LeetCode-Golang] 15. 三数之和的相关文章

15.三数之和——LeetCode

1 package threesum; 2 3 import java.util.ArrayList; 4 import java.util.Arrays; 5 import java.util.List; 6 7 /** 8 * 给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 9 * 注意:答案中不可以包含重复的三元组 10 * 11 * 主要的思想:如果采用暴力法的话,时间复杂

leetcode(15)三数之和+去重

三数之和 解题思路:排序+双指针 class Solution { public List<List<Integer>> threeSum(int[] nums) { List<List<Integer>> result = new ArrayList<List<Integer>>(); if(nums.length<=2){ return result; } List<Integer> item = null; A

15. 三数之和(筛选数据)

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 注意:答案中不可以包含重复的三元组. 例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2]] 1/** 2 * @param {number[]} nums 3 * @return {number[][]} 4 */ 5 6var th

LeetCode 15. 三数之和(3Sum)

题目描述 给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 注意:答案中不可以包含重复的三元组. 例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ] 解题思路 首先对数组从小到大排序,从一个数开始遍历,若该数大于0,后面的数不可能与其相加和为0,所以跳过:否则该数可能是满足要求

leetcode 15.三数之和

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 注意:答案中不可以包含重复的三元组. 例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2]] class Solution {public:    vector<vector<int>> threeSum(vector&l

[Leetcode 15]三数之和 3 Sum

[题目] Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero. Note: The solution set must not contain duplicate triplets. Example: Given array nu

Python版[leetcode]15. 三数之和(难度中等)

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组. 注意:答案中不可以包含重复的三元组. 示例: 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2]] 这道题我感觉是有点难度的,一般都是2个数做比较,这次有3个数参与了,没有想到什么好的方法,后来直接找的参考 排序 + 双指针 本题的难点在于如

leetcode新年病房暴乱康复计划 15. 三数之和 JS解法

var threeSum = function(nums) { var ans = []; var nums = nums.sort(function(a,b){return a - b}); var flag = 0; while(flag < nums.length - 2){ if (nums[flag] > 0){ break; } var l = flag + 1,r= nums.length - 1; while(l < r){ var res = nums[flag] +

15. 三数之和 (思维)

题目连接: https://leetcode-cn.com/problems/3sum/ 题目大意: 中文题目 具体思路: 使得 a +b +c = 0,这个等式成立有如下情况 1. 都为0 2. 一正两负/两正一负 3. 一正一负一零 观察之后,除了0这种情况,都会存在一个负数,我们可以第一步枚举负数,枚举完之后,再从剩下的数中凑得等式成立 我们将所有的数进行排序之后,第一部分枚举负数,第二部分通过两个指针来凑数 关于去重,如果负数都相同,这个好判断. 需要考虑的是剩下的两个数的去重,我们可以