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;
};
全部评论

相关推荐

Twilight_m...:表格简历有点难绷。说说个人看法: 1.个人基本情况里好多无意义信息,什么婚姻状况、健康状况、兴趣爱好、户口所在地、身份证号码、邮政编码,不知道的以为你填什么申请表呢。 2.校内实践个人认为对找工作几乎没帮助,建议换成和测开有关的项目,实在没得写留着也行。 3.工作经历完全看不出来是干什么的,起码看着和计算机没啥关系,建议加强描述,写点你在工作期间的实际产出、解决了什么问题。 4.个人简述大而空,看着像AI生成,感觉问题最大。“Python,C,C++成为我打造高效稳定服务的得力工具”、“我渴望凭借自身技术知识与创新能力,推动人工智能技术的应用发展,助力社会实现智能化转型”有种小学作文的美感。而且你确定你个人简述里写的你都会嘛?你AI这块写的什么“深入研究”,发几篇顶会的硕博生都不一定敢这么写。而且你AI这块的能力和软测也完全无关啊。个人简述建议写你对哪些技术栈、哪些语言、哪些生产工具的掌握,写的有条理些,而且最好是和测开强相关的。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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