双指针模板

最新推荐文章于 2024-10-13 16:52:53 发布

蕉蕉你要A起来

最新推荐文章于 2024-10-13 16:52:53 发布

阅读量104

点赞数 1

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

本文链接：https://blog.csdn.net/feijiaojiao/article/details/128020945

版权

三数之和双指针排序算法 Python 数组处理

关键词由CSDN通过智能技术生成

python 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

三数求和为0问题，循环一个数，另两个用双指针while循环

nums = [-1,0,1,2,-1,-4] 
def threeSum(nums):
    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
        if(i>0 and nums[i]==nums[i-1]):
            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