[BOJ 5568번] 카드 놓기(Python)

📝 문제

💡 상근이는 카드 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이, 둘째 줄에 k가 주어진다.
셋째 줄부터 n개 줄에는 카드에 적혀있는 수가 주어진다.

⚙️ 출력 : 첫째 줄에 상근이가 만들 수 있는 정수의 개수를 출력한다.

 

문제링크

5568번: 카드 놓기

---

✅ 풀이

from itertools import permutations

n = int(input())
k = int(input())
cards = []
for i in range(n):
    cards.append(input())

numbers = list(permutations(cards, k))
answer = []

for items in numbers:
    temp = ''
    for item in items:
        temp += item
    answer.append(temp)

print(len(set(answer)))

💡 카드들의 순서가 중요하기 때문에, 간단히 순열을 사용하였다.
주어진 카드들의 순서를 고려해 k개 뽑은 경우들을 순열을 통해 리스트로 만들고,
각 케이스들을 합쳐서 하나의 숫자로 만들어주었다.
이후, 숫자들 중 중복되지 않은 것들의 갯수를 세어 마무리했다.