[백준][코딩연습/백트래킹]#15652 N과 M(4)

www.acmicpc.net/problem/15652

15652번: N과 M (4)

문제

자연수 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)