[Python] | permutations 사용법 | combinations 사용법 | 순열 | 조합

1. 정의

순열 : 몇 개를 골라 순서를 고려해 나열한 경우의 수

조합 : 몇 개를 골라 순서를 고려하지 않고 나열한 경우의 수

2. 선언

import itertools

arr = ['A', 'B', 'C']
nPr = itertools.permutations(arr, 2) ##순열
#[('A', 'B'), ('A', 'C'), ('B', 'A'), ('B', 'C'), ('C', 'A'), ('C', 'B')]

nCr = itertools.combinations(arr, 2) ##조합
#[('A', 'B'), ('A', 'C'), ('B', 'C')]

3. 기본 매소드

4. 활용

https://www.acmicpc.net/problem/14889

14889번: 스타트와 링크

import sys
sys.stdin = open('input.txt', 'r')

# N 은 짝수이며, 두 팀으로 나눠야 한다
# 능력치
# 능력치의 차이를 최소로

n = int(input())
arr = [list(map(int, input().split())) for _ in range(n)]

member = {ni for ni in range(n)}
answer = 1000000
import itertools

for teamA in list(itertools.combinations(member, n//2)):
    scoreA, scoreB = 0, 0
    teamB = member-set(teamA)
    teamB = list(teamB)
    # print(teamA, teamB)


    for i in range(n//2-1):
        for j in range(i, n//2):
            scoreA += arr[teamA[i]][teamA[j]]
            scoreA += arr[teamA[j]][teamA[i]]
            scoreB += arr[teamB[i]][teamB[j]]
            scoreB += arr[teamB[j]][teamB[i]]
    answer = min(answer, abs(scoreA-scoreB))
    # print(teamA, teamB)
    # print(scoreA, scoreB)
print(answer)

---

참고1

https://velog.io/@dramatic/Python-permutation-combination-%EC%88%9C%EC%97%B4%EA%B3%BC-%EC%A1%B0%ED%95%A9

[Python] permutation, combination 순열과 조합

참고2

https://9kilometer.github.io/record/python-useful-things-for-algorithm-solve/

[PYTHON] 코딩 테스트에서 쓸만한 기능들 모음