文心一言 Python编程之

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

示例1: 输入:nums=[-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]

示例2: 输入:nums = [] 输出:[]

示例3: 输入:nums = [0] 输出:[]

文心一言编程如下(自带注释)

python 复制代码
def three_sum(nums):
    if len(nums) < 3:
        return []

    nums.sort()  # 对数组进行排序
    result = []

    # 遍历数组,固定第一个元素
    for i in range(len(nums) - 2):
        # 避免重复的三元组
        if i > 0 and nums[i] == nums[i - 1]:
            continue

            # 双指针法,在剩余部分寻找和为-nums[i]的两个数
        left = i + 1
        right = len(nums) - 1

        while left < right:
            total = nums[i] + nums[left] + nums[right]
            if total < 0:
                # 和太小,移动左指针
                left += 1
                # 避免重复的三元组
                while left < right and nums[left] == nums[left - 1]:
                    left += 1
            elif total > 0:
                # 和太大,移动右指针
                right -= 1
                # 避免重复的三元组
                while left < right and nums[right] == nums[right + 1]:
                    right -= 1
            else:
                # 找到和为0的三元组
                result.append([nums[i], nums[left], nums[right]])
                left += 1
                right -= 1
                # 避免重复的三元组
                while left < right and nums[left] == nums[left - 1]:
                    left += 1
                while left < right and nums[right] == nums[right + 1]:
                    right -= 1

    return result


# 示例1
nums1 = [-1, 0, 1, 2, -1, -4]
print(three_sum(nums1))  

# 示例2
nums2 = []
print(three_sum(nums2))  

# 示例3
nums3 = [0]
print(three_sum(nums3))  
相关推荐
没有不重的名么几秒前
在Pytorch中使用Tensorboard可视化训练过程
人工智能·pytorch·python
程序媛徐师姐1 分钟前
Python基于Django的全国二手房可视化分析系统【附源码】
python·django·全国二手房可视化分析系统·二手房可视化分析系统·二手房房价可视化·二手房可视化·二手房房价可视化分析系统
互联网打工人no17 分钟前
.NET8 依赖注入组件
开发语言·c#·.net·ioc
道剑剑非道13 分钟前
QT开发技术【QT实现桌面右下角消息】
开发语言·数据库·qt
fs哆哆14 分钟前
在VB.net和VBA中,自定义函数GetTargetSheet()返回工作表对象
java·开发语言·前端·javascript·ecmascript
KyollBM14 分钟前
【CF】Day44——Codeforces Round 908 (Div. 2) C + Codeforces Round 1020 (Div. 3) DE
算法
努力学习的小廉18 分钟前
我爱学算法之—— 二分查找(上)
开发语言·c++·算法
东木月19 分钟前
Python解析地址中省市区街道
开发语言·python
小彭律师26 分钟前
SinSR模型剪枝实验报告
人工智能·算法·剪枝
意.远33 分钟前
PyTorch卷积层填充(Padding)与步幅(Stride)详解及代码示例
人工智能·pytorch·python·深度学习