LeetCode——中级算法——数组和字符串——三数之和

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

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]


思路

如果这题用暴力求解,三层循环,不仅慢,而且会出现重复的三元组。
而三元组内部允许出现重复元素,例如 [-1, -1, 2],因此也不能使用set集合。
因此想到,先固定一个数,然后在用双指针在其余数中寻找和为其相反数的两个数,但怎样才能避免出现重复的三元组呢?
答案是 先将数组排序。将数组排序后,重复的数直接跳过。
所以第一步,排序:

nums.sort((a,b) => a - b);

注意默认的排序sort()会把 -2 排在 -4 前面,造成 [ -2 , -3 , -4, 0, 2, 3, 3, 4] 这种情况。
先固定一个负数,然后在后面的数中找两个数。
使用k作为固定数的索引,i、j 作为双指针从 k + 1 和 数组末尾开始。target作为nums[k]的相反数

let target = 0 - nums[k]
let i = k + 1;
let j = nums.length - 1;

当 i < j 时循环

如果i、j所指的数相加等于target,说明可以组成三元组,加入result。
然后把 i 往后移到下一个不相同的数,把 j 往前移到下一个不相同的数。
如果小了,移 i; 大了就移 j。

总代码:

/** * @param {number[]} nums * @return {number[][]} */
var threeSum = function(nums) {
  nums.sort((a,b) => a - b);
  let result = [];
  for (let k = 0; k < nums.length; k ++) {
    if (nums[k] > 0) break;
    if (nums[k] === nums[k-1]) continue;
    let target = 0 - nums[k];
    let i = k + 1,
        j = nums.length - 1;
    while (i < j) {
      if (nums[i] + nums[j] === target) {
        result.push([nums[k], nums[i], nums[j]])
        while (i < j && nums[i] === nums[i + 1]) {
          i ++;
        }
        while (i < j && nums[j] === nums[j - 1]) {
          j --;
        }
        ++i; --j;
      } else if (nums[i] + nums[j] < target) {
        ++i;
      } else {
        --j;
      }
    }
  }
  return result;
};
全部评论

相关推荐

头像
01-12 14:44
已编辑
百度_高级研发工程师
今天看到了某平台攻击牛友的帖子,段段今天打算为牛友们说句话,我们的努力到底有没有意义。&nbsp;(原文复述:感觉牛客就是当年那群做题区毕业了开始找工作还收不住那股味,颇有一种从年级第一掉到年纪第二后抱怨考不上大学的区味)&nbsp;&nbsp;粗鄙,无礼,傲慢,攻击,在这里我没有看到任何有用的分析,我只看到了屁股决定脑袋的攻击,我只看到了嫉妒和眼红。一、去医院不看病你去逛街吗&nbsp;去医院你不去看病你去逛街吗?去加油站不加油你去抽烟吗?去部队你不训练战斗技能你去养老吗?来牛客你不努力求职你来干什么来了。&nbsp;牛客本身就是个求职平台,大家分享有用的知识,分享面经,分享offer,分享求职经验的,来牛客不就干这个来了吗?有什么问题吗?...
给个好点的工作吧啊啊...:不知道我看的是不是和博主同样的帖子,我记得原帖是表达的是有些匿名老是发几十万的offer侮辱价,然后就有牛友觉得凡尔赛了导致后面的评论有些偏激。我觉得这个最近闫学晶那个事情有点类似了,她说他儿子一年只能赚七八十万家庭生活都难以为继,不说普通家庭,多少大厂的程序员都赚不到这个数字,大部分家庭看到这种发言肯定会难受的一p,生活的担子又这么重,人都是需要发泄情绪的,互联网就是个极佳的载体,所以很多人直接就喷她了,人在情绪发泄的时候是不思考的,否则就不叫发泄了。然后还有一个点,段哥假定了这些喷的人全都是“躺平的”,这点可能有失偏颇,很多人一直在努力,但是始终缺乏天时地利人和的某一个条件,这点相信段哥找工作的过程中深有体会。绝大部分人都以结果的失败去否认了努力的全过程,可能只是别人努力的方向错了。就像一次面试,可能你准备了很久,结果面试官就是比较奇葩,一直问没有学习到的领域或知识点,然后有人凭一个挂掉的结果就直接给你扣了一个“躺平”的帽子,觉得挂掉是你不够努力,您心里滋味如何?再说点近点的,我也是od,多少同事深夜无偿加班,涨过一分工资吗?多少外包的技术大牛因为学历被困在外包,连od都进不去,这些人难道不努力吗?只是限制与生活、公司制度等等之类的无奈罢了。说到努力,又想到李家琦79元眉笔事件,这么多年有没有认真工作?有没有涨工资?他嘴里说出来是那么的理所当然,打工牛马都知道“任劳任怨”,“认真工作”真能涨工资?只干活不发声就等着被摘果子吧,企业里永远都是“汇报杰出者”升的最快(当然不是所有企业),这种事情相信段哥包括我甚至大部分od都经历过。最近辞职回老家,和老爸散步每次他都会感慨街上的蔬菜小贩多不容易,他们晚上就窝在那种三轮小货车的驾驶室里,腿都伸不直,我们这里晚上零下了,只盖一条薄毛毯,始终舍不得住我们镇上几十块的酒店,因为一车蔬菜就赚几百块顶多一千而且要卖好久,这样的例子还有太多了。这种芸芸众生可能辛苦了一天之后,打开手机看到网上的凡尔赛发言,跟风喷了几句发泄情绪,我觉得这种人不应该扣上“躺平”的帽子。我觉得大部分正常人都是努力的,或者曾经努力过,但世界上有太多努力解决不了的无奈了,甚至说你都没有那个努力的机会,不过正因如此,才显得坚持不懈的努力奋斗之人的难得可贵,认清生活的真相后仍然热爱生活,敢于直面现实的淋漓。
段段STEADY觉醒与突...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务