python--三数之和

三数之和

'''
给定一个包含——n——个整数的数组'num',判断’num'中是否存在三个元素a,b,c是的a+b+c=0
找出所有满足条件且不重复的三元数组
'''

def day01(nums):
    result=[]
    for idx0,i in enumerate(nums):
        a=i
        for idx1,ii in enumerate(nums):
            if idx0!=idx1:
                b=ii
                for idx2,iii in enumerate(nums):
                    if idx2!=idx0 and idx2!=idx1:
                        c=iii
                        if a+b+c==0:
                            if sorted([a,b,c]) not in result:
                                result.append(sorted([a,b,c]))
    return result
if __name__ == '__main__':

    nums = [-1, 0, 1, 2, -1, -4]
    result=day01(nums)
    print(result)

猜你喜欢

转载自blog.csdn.net/qestion_yz_10086/article/details/107656755