from itertools import combinations
def solution(number):
return sum(not sum(c) for c in combinations(number, 3))
조합 라이브러리를 이용하는 동작
코딩테스트에서 이런 라이브러리가 허용되면 과감하게 사용해야한다.
def solution(number):
answer = 0
l = len(number)
for i in range(l-2):
for j in range(i+1, l-1):
for k in range(j+1, l):
# print(number[i],number[j],number[k])
if number[i]+number[j]+number[k] == 0:
answer += 1
return answer
내가 생각하는 좋은 풀이이다.
탐색하며 O(n)을 유지하는 모습..!