Python教程

python刷题--N数之和问题(双指针+剪枝)

本文主要是介绍python刷题--N数之和问题(双指针+剪枝),对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

1.两数之和(双指针)

这题前面已经做过,当时是用哈希表做的,时间复杂度为N

但如果换一种思路,用今天学的双指针来做,虽然在时间复杂度上不降反增(因为排序的复杂度为NlogN)但理解起来十分简单清晰。(注:对于三数四数N数之和问题来说,双指针算法相当于将最内部的两层循环优化成一层。)

思路:即先排序,得到一个有序的列表,一个指针指向列表开头,一个指向末尾,对于每一个状态而言,如果大于目标值则让右指针左移,小于目标值则让左指针右移。该方法适用于求无重复的解。

个人题解:

# 注释1
class Solution:
    def twoSum(self, nums: List[int], target: int):
        res = []
        i, j = 0, len(nums)-1
        nums.sort()
        for i in range(len(nums)):
            if i > 0 and nums[i-1] == nums[i]:
                continue
            while i < j and nums[i] + nums[j] > target:
                j -= 1
            if i == j:  # 注释2,3
                break
            if nums[i] + nums[j] == target:
                res.append([nums[i], nums[j]])
        return res

注释1:# 注意这不是题目:两数之和的解,这代码的条件是跟后面的三数之和四数之和相同时才能成立(必须要求只返回目标值而不是返回索引)

注释2:# 能导致i=j的只有两种情况。一:由内层循环得到,那么当前状态的前一个状态nums[i]+nums[j] 是大于target的,此时期待减少两个数的和,但是控制减小的指针j已经无法向左移,所以循环结束二:由外层循环得到,那么当前状态的前一个状态nums[i]+nums[j]是小于target的,此时期待增大两个数的和,但是控制增大的指针i已经无法向右移,循环结束

注释3:在上述情况发生时(期待增大的两个数之和,但是控制增大的指针i无法向右移)。那么可不可以让j往右移动(回头)呢?

答:是不行的。看上去j向右移动,和这个状态的i是一组全新的i,j组合,是没有遍历过的,那为什么不能这样呢?

我们来假设这样的情况成立。。假设在移动之前i和j的位置分别为i0和j0那么j0向右移动后移到某个j1的位置,此时nums[i0]+nums[j1]的值是大于等于0(或者目标值的),假设i在最开始的位置是i1,那么当j第一次运动到j1的位置时,i在i1的位置,这个时候的nums[i1]+nums[j1]是小于nums[i0]+nums[j1]的,而j1运动到j0是为了使整体值减小,那么第一次的nums[i1]+nums[j1]一定就是大于0,所以对比两个状态,i1和j1这个状态下已经大于0,那么就不会再让i1增大得到i0,所以与我们一开始设定好的值大就减小,值小就增大的思想不符,假设不成立。

2.三数之和

题目:https://leetcode-cn.com/problems/3sum/

官方题解:

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        nums.sort()
        ans = list()
        
        # 枚举 a
        for first in range(n):
            # 需要和上一次枚举的数不相同
            if first > 0 and nums[first] == nums[first - 1]:
                continue
            # c 对应的指针初始指向数组的最右端
            third = n - 1
            target = -nums[first]
            # 枚举 b
            for second in range(first + 1, n):
                # 需要和上一次枚举的数不相同
                if second > first + 1 and nums[second] == nums[second - 1]:
                    continue
                # 需要保证 b 的指针在 c 的指针的左侧
                while second < third and nums[second] + nums[third] > target:
                    third -= 1
                # 如果指针重合,随着 b 后续的增加
                # 就不会有满足 a+b+c=0 并且 b<c 的 c 了,可以退出循环
                if second == third:
                    break
                if nums[second] + nums[third] == target:
                    ans.append([nums[first], nums[second], nums[third]])
        
        return ans

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

思想一样,就是在外层再套一个循环。

3.四数之和

题目:https://leetcode-cn.com/problems/4sum/

整体思路一样,题解可以看看,多了个剪枝操作(通过判断情况筛选掉不必要的循环)

官方题解:

https://leetcode-cn.com/problems/4sum/solution/si-shu-zhi-he-by-leetcode-solution/https://leetcode-cn.com/problems/4sum/solution/si-shu-zhi-he-by-leetcode-solution/

4.N数之和 

道理一样,就是在外层套循环。然后将最内的两层循环改为双指针操作。

这篇关于python刷题--N数之和问题(双指针+剪枝)的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!