Python基础练习（三）：0的组合

题目描述

题目内容：

给定一个包含若干个整数（可能存在重复整数）的列表，判断其中是否存在三个元素a,b,c，使得a+b+c=0？找出所有满足条件且不重复的这样的三个数的组合。

输入格式:

共一行，列表中元素以空格隔开。

输出格式：

共一行，为不重复组合的个数，不存在这样的组合就输出0

输入样例：

-1 0 1 2 -1

输出样例：

2

（注：两个组合是-1，-1，2和-1，0，1）

时间限制：500ms内存限制：32000kb

解题思路

1. 一个列表找出三个元素的组合，因此联想到遍历的方法
2. 要求不能重复，我们可以对每一次遍历的个数进行限制，即每次遍历除上次遍历的初始值以外的所有数；同时考虑到原始数据的无序性，可能出现周期循环情况，因此可以在遍历之前对列表进行排序。
3. 找出组合后我们可以把它作为一个整体放到预设的一个空集合中，最后输出的组合次数即为集合的长度。（利用集合中不能出现相同元素这一特性）

解题代码

a = sorted(list(map(int,input().split())))
l = len(a)
b = set()
for i in range(l):
    for j in range(i+1,l):
        for k in range(j+1,l):
            if a[i]+a[j]+a[k]==0:
                b.add((a[i],a[j],a[k]))
print(len(b))