15. 三数之和

一、题目原型:

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

二、题目意思剖析:

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

三、解题思路:

3.1.首先想到的最简单最傻的做法是三层循环,时间复杂度为O(n的三次方)。。。想想都觉得耗时。。。非常不建议,只是一个思路而已。

func threeSum(_ nums: [Int]) -> [[Int]] {
    
    var newNums: [Int] = []
    var haha:[[Int]] = []
    for i in 0..<nums.count {
        
        for j in i+1..<nums.count {
            
            for k in j+1..<nums.count {
                
                if nums[i] + nums[j] + nums[k] == 0 {
                    newNums.append(nums[i])
                    newNums.append(nums[j])
                    newNums.append(nums[k])
                    newNums.sort()
                    if haha.contains(newNums) {
                        newNums.removeAll()
                        continue
                    }
                    haha.append(newNums)
                }
                newNums.removeAll()
            }
        }
    }
    return haha
}

3.2.经过耗时太长的惨痛经历,痛定思痛,我觉得从题目“三数之和=0”入手。排除三个数都为0的情况,也就是[0,0,0];请他情况必定是有一个负数的。所以我们可以首先就排个序。

//第一步:排序
比如刚开始是[-1, 0, 1, 2, -1, -4]
排序后变成是[-4, -1, -1, 0, 1, 2]

排序完,是不是觉得思路有了,对的。我们只需要遍历前面的负数和0,也就是>0的我们统统不要了。我们需要拿到第一个负数。

for i in 0..<MutNums.count {
    if (MutNums[i] > 0) {
        break;
    }
    // MutNums[i]就是一个负数
}

拿到第一个负数后,我们只需要再拿到后面两个数,与之相加=0即可。后面两个数我们用两个指针来表示,j和k,一个是从左边往右边走,一个是从最右边往前走。有些思想就是想通的,之前的两数之和就是这种思路。先排好序,左边一个left指针,右边一个right指针,如果left+right > result,那么right往左移,如果left+right < result,那么说明相加得数不够,就需要left往右移。找到相加之和相等的就行。此题,也是如此。

let target = 0 - MutNums[i]; // 这个就是动态的reslut
var j = i+1, k = MutNums.count - 1
if (MutNums[j] + MutNums[k] == target) {
    //这就回到了两数之和那个简单的问题。
}

不过在这里要注意一个问题,就是题目中提到的“答案中不可以包含重复的三元组”,所以我们需要把相同的数字过滤掉。
其实很简单,只需要往下或者往前走一位,其他什么都不要做。

while j < k && MutNums[j] == MutNums[j+1] { 
    j = j + 1
}

贴出代码

func threeSum(_ nums: [Int]) -> [[Int]] {
    var MutNums: [Int] = nums
    var newNums: [Int] = []
    var haha:[[Int]] = []
    // 1.排序 对于MutNums[i]来说,我们只需要负数和0,因为三个数之和为0,一定是有一个数为负数的,当然除去三个数都为0的情况。所以,我们取非正数。
    MutNums.sort()
    for i in 0..<MutNums.count {
        if (MutNums[i] > 0) {
            break;
        }
        // 如果两个数字相同,我们直接跳到下一个循环。
        if (i > 0 && MutNums[i] == MutNums[i-1]) {
            continue
        }
        let target = 0 - MutNums[i];
        var j = i+1, k = MutNums.count - 1
        while j < k {
            // 2.找到后面的两个与MutNums[i]对应的数字
            if (MutNums[j] + MutNums[k] == target) {
                newNums.append(MutNums[i])
                newNums.append(MutNums[j])
                newNums.append(MutNums[k])
                haha.append(newNums)
                newNums.removeAll()
                // 如果两个数字相同,我们直接跳到下一个循环。
                while j < k && MutNums[j] == MutNums[j+1] {
                    j = j + 1
                }
                // 如果两个数字相同,我们直接跳到下一个循环。
                while j < k && MutNums[k] == MutNums[k-1] {
                    k = k - 1
                }
                // 否则就往中间靠拢
                j = j + 1;k = k - 1
            }else if (MutNums[j] + MutNums[k] < target) {
                // 如果后面两数相加小于target,说明左边还得往右移
                j = j + 1
            }else {
                // 如果后面两数相加大于target,说明右边就要往左移
                k = k - 1
            }
        }
    }
    return haha
}
运行数组
[13,9,1,12,-7,-12,7,3,9,6,-7,4,9,5,5,-7,4,11,1,-2,12,3,-12,-15,0,-12,-6,-1,7,-5,-4,-3,12,4,-14,-10,-1,8,1,-6,-1,9,13,-14,-1,-5,-6,-12,-8,2,2,11,13,-3,11,-2,1,-10,4,-15,-8,7,-11,11,-4,-10,-13,3,5,3,12,11,-11,2,12,3,13,13,-2,12,-7,-15,8,-9,-10,-4,-4,6,1,-15,-2,0,-1,2,-3,10,-1,-9,-10,-11,1,-13,-15,5,-3,5,-7,-5,-5,6,14,3,-1,7,1,-4,-12,12,-13,-4,4,0,3,-12,9,-15,6]

耗时对比:
1.三层循环 347.76ms
2.滑尺优化方案 2.64ms
话不多说,感受下这个天壤之别。

四、小结

总提交数
提交结果

有任何疑问都可以留言,非常乐意一起探讨。😄
个人博客地址

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,311评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,339评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,671评论 0 342
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,252评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,253评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,031评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,340评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,973评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,466评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,937评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,039评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,701评论 4 323
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,254评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,259评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,485评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,497评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,786评论 2 345

推荐阅读更多精彩内容