15654번: N과 M (5)
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열
www.acmicpc.net
문제
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다.
- N개의 자연수 중에서 M개를 고른 수열
입력
첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.
출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
수열은 사전 순으로 증가하는 순서로 출력해야 한다.
코드
- 백트래킹 이용 풀이
n, m = map(int, input().split())
arr = sorted(list(map(int, input().split())))
lst = []
visited = [False] * (n+1)
def dfs(idx):
if idx > n:
return
if len(lst) == m:
print(*lst)
return
for i in range(n):
if not visited[i]:
lst.append(arr[i])
visited[i] = True
dfs(i+1)
lst.pop()
visited[i] = False
dfs(0)
dfs를 이용하여 재귀함수를 구현해서 풀이하였다.
- dfs()
- idx는 arr의 인덱스를 뜻하는데 인덱스의 범위를 넘으면 리턴한다.
- 우리가 원하는 개수만큼 숫자를 뽑았다면 뽑은 숫자들을 출력하고 리턴해준다.
- 조합이 아닌 순열을 구하는 것이므로 순서도 상관이 있다. 그러므로 for문의 범위는 0부터로 한다.
- itertools 라이브러리의 permutations 이용 풀이
from itertools import permutations
n, m = map(int, input().split())
arr = sorted(list(map(int, input().split())))
for comb in permutations(arr, m):
print(*comb)
'문제풀이 > 백트래킹' 카테고리의 다른 글
[Python/파이썬] 백준 10971번 외판원 순회 2 (1) | 2023.03.11 |
---|---|
[Python/파이썬] 백준 15655번 N과 M (6) (0) | 2023.03.10 |
[Python/파이썬] 백준 15652번 N과 M (4) (0) | 2023.03.10 |
[Python/파이썬] 백준 15651번 N과 M (3) (0) | 2023.03.10 |
[Python/파이썬] 백준 1182번 부분수열의 합 (0) | 2023.03.09 |