LeetCode&Python 58.四数之和 (推广至N数之和)

描述

给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。

样例

例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:

(-1, 0, 0, 1)

(-2, -1, 1, 2)

(-2, 0, 0, 2)

注意事项

四元组(a, b, c, d)中,需要满足a <= b <= c <= d

答案中不可以包含重复的四元组。


V1: 利用三数之和的思想再加一层循环,这种方法的时间复杂度是O(n3),会超出时间限制。

class Solution:

    def fourSum(self, numbers, target):

        p = []

        numbers.sort()

        for i in range(len(numbers)-3):

            t1 = target - numbers[i]

            for j in range(i+1, len(numbers)-2):

                t2 = t1 - numbers[j]

                q = {}

                for k in range(j+1, len(numbers)):

                    t3 = t2 - numbers[k]

                    if numbers[k] in q:

                        a = numbers[i]

                        b = numbers[j]

                        c = t3

                        d = numbers[k]

                        l = (a,b,c,d)

                        if l not in p:

                            p.append(l)

                    else:

                        q[t3] = k

        return p                           

V2: 将四个数字分成两部分,将第一部分两个数字之和和它们的下标存入字典,用target减去剩下两个数字之和设为t2,在字典中寻找t2,找到后对比它们的下标,排序后返回结果。这种方法的时间复杂度是O(n2),空间复杂度也是O(n2),但提交的时候也会超过时间限制。

class Solution:

    def fourSum(self, numbers, target):

        p = []

        q = {}

        numbers.sort()

        for i in range(len(numbers)):

            for j in range(i+1, len(numbers)):

                t1 = numbers[i]+numbers[j]

                if t1 not in q: #这条语句其他博客里有用q.keys(),.keys()返回的是一个list,判断一个元素是否在list里是O(n)的,而判断元素是否在dict里是O(1)的

                    q[t1] = [(i,j)]

                else:

                    q[t1].append((i,j))

        for m in range(len(numbers)):     

            for n in range(m+1, len(numbers)-2):

                t2 = target - numbers[m] - numbers[n]

                if t2 in q:

                    for index in q[t2]:

                        if index[0] > n: #此处为了排列顺序,但是四个数的组合会重复两次,所以要去重

                            l = (numbers[m],numbers[n],numbers[index[0]],numbers[index[1]])

                            if l not in p: #如果p声明的是set()则不用判断,因为集合会去重

                                p.append(l)

        return p

V3: 我想这几道题的意思不止是解决单个的题目,而是可以推广到解决N数之和,在这里借鉴了前辈的代码加以学习。采用了递归的思想,先判断N是否等于2,如果比N大那么用递归先固定N-2个数,再判断两数之和。

class Solution:

    def fourSum(self, numbers, target):

        numbers.sort()

        results = []

        self.NSum(numbers, target, 4, [], results)

        return results


    def NSum(self, numbers, target, N , result, results):

        if len(numbers) < N or N < 2:return

        if N == 2:

            l = 0

            r = len(numbers)-1

            while l < r:

                if numbers[l] + numbers[r] == target:

                    results.append(result +[numbers[l],numbers[r]])

                    l+=1

                    r-=1

                    while numbers[l] == numbers[l-1]:

                        l+=1

                    while numbers[r] == numbers[r+1]:

                        r-=1

                elif numbers[l] + numbers[r] < target:

                    l+=1

                else:

                    r-=1

        else:

            for i in range(0,len(numbers)-N+1):

                if numbers[i]*N > target or numbers[-1]*N < target:

                    break

                if i == 0 or i > 0 and numbers[i-1] != numbers[i]:

                    self.NSum(numbers[i+1:], target-numbers[i], N-1, result+[numbers[i]],results) #在这里递归

        return results

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

推荐阅读更多精彩内容

  • 我们家姐弟五人,除了弟弟在外地工作外,其余三个姐们加上我一共四个人都在本市附近上班,周日的时候,如果大家都有空,我...
    三门峡014张丽娜阅读 223评论 1 5
  • 前几天一直是下雨,并不是很热。而今天下午很热,现在我有机会吃冰淇淋和吹冷气了。 我一把推开冰箱门,准备...
    施逸辰阅读 348评论 2 5
  • 昨晚我写完日记都过了12点啦,大家又开车来到了这次旅游的最后一站――威海。登上了侨乡号游轮,入住在游轮上的客房。...
    张轩赫阅读 195评论 0 1