每日一题——n数之和

苏炘

于 2023-02-10 18:33:28 发布

阅读量154

点赞数

分类专栏：数据结构与算法文章标签：算法数据结构 python

本文链接：https://blog.csdn.net/weixin_44902962/article/details/128969343

版权

数据结构与算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏

n数之和是一系列题目，题设为在一个数组中，找出全部的不重复的n元组，使得这n个数的和为0（或者为target，原理相同）

例如：两数之和

原题目：力扣

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那两个整数并返回它们的数组下标。

解法1：使用哈希表来将搜索target-num的时间复杂度降至 $O(1)$ ，整体时间复杂度 $O(n)$ ，空间复杂度 $O(n)$ 。

def twoSum(nums,target):
    hash_map={}
    for i,num in enumerate(nums):
        if target-num in hash_map: #前面是不是有可以满足题意的数字？
            return [i,hash_map[target-num]]
        target[num]=i

解法2：先对表排序，然后使用双指针，时间复杂度 $O(nlog_2n)$ ，空间复杂度 $O(1)$ 。虽然时间复杂度提高了，但是这种思想是后续n数之和的基本思想。

def twoSum(nums,target):
    i,j=0,len(nums)-1
    nums.sort()
    while i<j:
        if nums[i]+nums[j]>target:
            j-=1
        elif nums[i]+nums[j]<target:
            i+=1
        else:
            return [i,j]

从两数之和到三数之和：

原题目：力扣

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

解法1：如果我们还是考虑哈希表，原本的寻找target-num 就变成了寻找当前元素前面和为-num的组合，时间复杂度为 $O(n^2)$ ，空间复杂度 $O(n)$ 。

class Solution:
    def threeSum(self, nums: List[int]):
        hash_nums = {}
        result_list = []
        nums.sort()
        if nums!=[] and nums[0] > 0:
            return []
        for ti, tnumi in enumerate(nums):
            hash_nums[tnumi] = ti #依旧是以空间换时间，让查找值为某个数的索引这一操作复杂度降到O(1)
        for i, numi in enumerate(nums):
            if i>0 and nums[i]==nums[i-1]: #如果小的数字都大于0，那后面的就更没希望了
                continue
            for j, numj in enumerate(nums[i + 1:]):
                if j>0 and nums[i+1+j]==nums[j+i]:
                    continue
                c_re = hash_nums.get(-(numi + numj))
                if c_re is not None and c_re > j + i+1:
                    result_list.append([numi, numj, -(numi + numj)])        
        return result_list

解法2：排序+双指针，定住一个坐标动另外两个。

class Solution:
    def threeSum(self, nums: List[int]):
        n=len(nums)
        res=[]
        if(not nums or n<3):
            return []
        nums.sort()
        res=[]
        for i in range(n):
            if(nums[i]>0):
                return res #如果小的元素都大于0，大的就更没希望了
            if(i>0 and nums[i]==nums[i-1]): #如果索引为i的元素和前一个相同，那就一定会找到相同的解，故跳过
                continue
            L=i+1
            R=n-1
            while(L<R): #从这里开始其实是类似于两数之和的方法
                if(nums[i]+nums[L]+nums[R]==0):
                    res.append([nums[i],nums[L],nums[R]])
                    while(L<R and nums[L]==nums[L+1]):
                        L=L+1
                    while(L<R and nums[R]==nums[R-1]):
                        R=R-1
                    L=L+1
                    R=R-1
                elif(nums[i]+nums[L]+nums[R]>0):
                    R=R-1
                else:
                    L=L+1
        return res

同理，我们有在四数之和，n数之和都有类似的做法，该类问题即可得到解决，时间复杂度为 $O(n^{n-1})$ 。