面试题-python3 个包含n个整数的数组nums,判断nums中是否存在三个元素,a,b,c,使得a+b+c=0

面试题

python面试题:
一个包含n个整数的数组nums,判断nums中是否存在三个元素,a,b,c,使得a+b+c=0?
找出所有和为0并且不重复的三元组。不可以包含重复的三元组。
如:nums=[-1,0,1,2,-1,-4]
输出[[-1,-1,2],[-1,0,1]];
如:nums=[],
输出[];
如:nums=[0]
输出[]

代码解决

遍历读取列表值,判断3个相加等于0

# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/

def get_targe(nums, targe=0):
    if len(nums) < 3:
        return []
    targe = 0
    result = []
    for i in range(len(nums)):
        for j in range(i+1, len(nums)):
            for k in range(j+1, len(nums)):
                # 判断等于目标值
                if nums[i]+nums[j]+nums[k] == targe:
                    temp = [nums[i], nums[j], nums[k]]
                    # 去除重复
                    if sorted(temp) not in result:
                        result.append(sorted(temp))
    return result

if __name__ == '__main__':
    nums = [-1, 0, 1, 2, -1, -4]
    result = get_targe(nums, targe=0)
    print(result)
posted @ 2021-05-14 15:38  上海-悠悠  阅读(2201)  评论(1编辑  收藏  举报