level1_삼총사

 

내가 제출한 코드

def solution(number):
    answer = 0
    len_number = len(number)
    for i in range(len_number):
        for x in range(i+1, len_number):
            for y in range(x+1, len_number):
                if number[i]+number[x]+number[y] == 0:
                    answer += 1
    return answer

 

3중 반복문을 사용해서 가능한 모든 숫자 조합을 생성하고, 각 조합의 합이 0인지 검사하여 조합의 수를 카운트하는 코드이다. 

 

다른 사람이 제출한 코드

def solution(number):
	from itertools import combinations
    cnt = 0
    for i in combinations(number,3):
    	if sum(i) == 0:
        	cnt += 1
    return cnt

 

combinations이라는 라이브러리가...있다니!!!!!!!!

 

combination (조합)

조합이란 서로 다른 n개 중에서 r개(n≥r) 취하여 조를 만들 때, 이 하나하나의 조를 n개 중에서 r개 취한 조합이라고 한다. 

조합은 순서를 고려하지 않기 때문에 [A, B, C]의 리스트에서 2개의 원소를 골라 나열하면
[(A, B), (A, C), (B, C)] 가 나오게 된다. 조합은 (A, B)와 (B, A)는 같은 것으로 취급한다.

 

import itertools

arr = ['A', 'B', 'C']
nCr = itertools.combinations(arr, 2)
print(list(nCr))

# 결과 : [('A', 'B'), ('A', 'C'), ('B', 'C')]

 

 

조합을 봤으니... 순열을 안보고 갈 수가 없다.

 

permutation (순열)

순열이란 몇 개를 골라 순서를 고려해 나열한 경우의 수를 말한다. 즉, 서로 다른 n 개 중 r 개를 골라 순서를 정해 나열하는 가짓수이며 순열이라는 의미의 영어 ‘Permutation’의 첫 글자 P를 따서 nPr로 표시한다.

순열은 순서를 고려하기 때문에 [A, B, C]의 리스트에서 2개의 원소를 골라 순서를 정해 나열하면
[(A, B), (A, C), (B, A), (B, C), (C, A), (C, B)] 가 나오게 된다. 즉 순열에서는 (A, B)와 (B, A)는 다른 것이다.

 

import itertools

arr = ['A', 'B', 'C']
nPr = itertools.permutations(arr, 2)
print(list(nPr))

# 결과 : [('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')]

 

다음에는 이 라이브러리.. 무조건 써먹어야겠당!!!!