으음~ 각 학생한테 매겨진 숫자가 있고 주어진 number중 3명을 골라 그 합이 0이 되는 경우의 수를 반환하는 문제이다.
즉, 주어진 배열 number에서 3개씩 그룹을 지은다음 이들의 총합이 0인 경우를 세면되는데 그룹을 지을때 조합(=combinations)를 사용하면 된다!
from itertools import combinations
def solution(number):
answer = 0
tmp = list(combinations(number, 3))
for item in tmp:
if sum(item)==0:
answer += 1
return answer
'알고리즘 & 코딩테스트 > 코딩테스트' 카테고리의 다른 글
Programmers Lv.1 "최소직사각형" (0) | 2023.06.01 |
---|---|
Programmers Lv.1 "시저 암호" (0) | 2023.06.01 |
Programmers Lv.1 "예산" (0) | 2023.05.31 |
Programmers Lv.1 "이상한 문자 만들기" (0) | 2023.05.24 |
Programmers Lv. 1 "3진법 뒤집기" (0) | 2023.05.24 |