5568번: 카드 놓기
예제 1의 경우 상근이는 11, 12, 21, 112, 121, 122, 212를 만들 수 있다.
www.acmicpc.net
문제
상근이는 카드 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개 줄에는 카드에 적혀있는 수가 주어진다.
출력
첫째 줄에 상근이가 만들 수 있는 정수의 개수를 출력한다.
코드
from itertools import permutations
n = int(input())
k = int(input())
cards = []
for _ in range(n):
cards.append(input())
nums = set()
for arr in list(permutations(cards, k)):
nums.add(int(''.join(arr)))
print(len(nums))
'문제풀이 > 완전탐색' 카테고리의 다른 글
[Python/파이썬] 백준 1025번 제곱수 찾기 (0) | 2024.01.11 |
---|---|
[Python/파이썬] 백준 16439번 치킨치킨치킨 (0) | 2023.04.27 |
[Python/파이썬] 백준 1969번 DNA (0) | 2023.04.27 |
[Python/파이썬] 백준 18511번 큰 수 구성하기 (0) | 2023.04.26 |
[Python/파이썬] 백준 15721번 번데기 (0) | 2023.04.26 |