(Python) - BOJ(5568번) : 카드 놓기

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

5568번: 카드 놓기

---

• 해설 : 

상근이는 카드 n(4 ≤ n ≤ 10)장을 바닥에 나란히 놓고 놀고있다. 각 카드에는 1이상 99이하의 정수가 적혀져 있다. 상근이는 이 카드 중에서 k(2 ≤ k ≤ 4)장을 선택하고, 가로로 나란히 정수를 만들기로 했다. 상근이가 만들 수 있는 정수는 모두 몇 가지일까?

 

예를 들어, 카드가 5장 있고, 카드에 쓰여 있는 수가 1, 2, 3, 13, 21라고 하자. 여기서 3장을 선택해서 정수를 만들려고 한다. 2, 1, 13을 순서대로 나열하면 정수 2113을 만들 수 있다. 또, 21, 1, 3을 순서대로 나열하면 2113을 만들 수 있다. 이렇게 한 정수를 만드는 조합이 여러 가지 일 수 있다.

 

n장의 카드에 적힌 숫자가 주어졌을 때, 그 중에서 k개를 선택해서 만들 수 있는 정수의 개수를 구하는 프로그램을 작성하시오.

 

 

 

---

• 풀이 :

정수들 중 N개를 선택하여 조합하는 모든 경우를 permutations를 사용하여 구한 후, 이 값들을 문자열로 취급해서 이어준다. 결괏값을 set처리하여 중복을 없에준 후 길이를 찾는다.

 

 

 

---

• 소스코드 : 

 

import sys
from itertools import permutations
input = sys.stdin.readline
 
if __name__ == '__main__':
    n = int(input())
    k = int(input())
    card = [input().strip() for _ in range(n)]
    ans = []
    for idx in permutations(card,k):
        value = ''
        for key in idx:
            value += key
        ans.append(value)
    print(len(set(ans)))