[코테] 순열 / 조합

Coding_Holic·2023년 9월 6일

코딩테스트 준비

목록 보기
8/12
itertools 라이브러리 암기!
import itertools

순열과 조합의 차이는 순서의 유무이다.

순열(Permutation)

서로 다른 n개 중 r개를 골라 순서를 정해 나열하는 가짓수
nPr=n!(nr)!_nP_r=\frac{n!}{\left(n-r\right)!}

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')]

조합(Combination)

서로 다른 N개 중에서 r개를 취하여 조를 만들 때 이 하나하나의 조를 n개 중에서 r개 취한 조합이라고 한다.
nCr=nPrr!=n!r!(nr)!_nC_r=\frac{_nP_r}{r!}=\frac{n!}{r!\left(n-r\right)!}

import itertools

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

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

itertools 기타 함수

zip

  • 동일한 개수로 이루어진 iterable한 객체들을 인수로 받아 묶어서 iterator로 반환
    z = zip([1, 2, 3], ('A', 'B', 'C'))
    print(next(z)) # (1, 'A')
    print(next(z)) # (2, 'B')
    print(next(z)) # (3, 'C')

all

  • 원소가 모두 참이면 True 아니면 False를 반환
  a = all([1, 2, 3]) # True
  a = all([0, 1, 2]) # False

any

  • iterable한 객체를 인수로 받아서 원소가 하나라도 참이면 True, 아니면 False를 반환
a = any([0, 1, 2]) # True
a = any([0, False, []] # False
profile
안녕하세용 개발에 미치고 싶은 초보 개발자입니다:)

0개의 댓글