coding test practice
[백준][코딩연습/백트래킹]#15652 N과 M(4)
예진잉구
2021. 1. 8. 02:26
15652번: N과 M (4)
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해
www.acmicpc.net
문제
자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
- 1부터 N까지 자연수 중에서 M개를 고른 수열
- 같은 수를 여러 번 골라도 된다.
- 고른 수열은 비내림차순이어야 한다.
- 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.
입력
첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
수열은 사전 순으로 증가하는 순서로 출력해야 한다.
N과 M 2, 3번 문제를 섞어 놓은 문제이다.
그래서 코드를 합친(?) 것이다. dfs의 2,3번째 인수가 같지만 이해하기 쉽기 위해 그냥 두 개로 적었다.
N,M = map(int, input().split())# 입력 받기
check = [False] * N # 사용했는지 안했는지 check 해 줄 배열, 길이는 숫자 개수 N과 같다.
answer = [0] * M # 정답 배열은 길이가 M 이다.
def dfs(idx, me, start):
check[me] = False # 사용한 나 자신도 포함해야 하니까 False로 바꾸고 시작
if idx == M:
for i in range(M):
print(answer[i], end=' ') # 원소들 출력할 때 엔터 안치고 띄어쓰기 함
print() # 배열 하나 출력하고 나면 엔터침
return # 돌아가라
else:
for i in range(start, N):
if check[i] == True: # 이미 사용했으면 그냥 넘어가기
continue
else: # 사용안했으면?
answer[idx] = i+1 # 정답 배열에 숫자를 추가한다. 인덱스는 0부터, 숫자는 1부터 시작이니까 i+1을 한다.
check[i] = True # 사용했다고 표시해 주기
dfs(idx+1, i, i) # 정답 배열의 다음 원소를 찾으러 재귀함수를 사용한다.
check[i] = False # 중요!) 다 사용했으면 False로 바꿔 사용하지 않았다고 표시 해 준다.
dfs(0,0,0)