python 实战39 0的组合

题目内容：

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

输入格式:

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

输出格式：

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

输入样例：

-1 0 1 2 -1

输出样例：

2

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

---

My answer
思路：为了避免重复，我把列表进行了排序，并用集合来装符合条件的元素。

n = list(map(int, input().split()))
n.sort()
s = set()
for i in range(len(n) - 2):
    for j in range(i + 1, len(n) - 1):
        for k in range(j + 1, len(n)):
            if n[i] + n[j] + n[k] == 0:
                s.add((n[i], n[j], n[k]))

print(len(s))

我们要更进一步python化，去掉几行

n = sorted(list(map(int, input().split())))
s = {(n[i], n[j], n[k]) for i in range(len(n) - 2) for j in range(i + 1, len(n) - 1) for k in range(j + 1, len(n)) if
     n[i] + n[j] + n[k] == 0}
if len(s) == 0:
    print(0)
else:
    print(len(s))