[백준][코딩공부/백트래킹]#15650 N과 M(2)

www.acmicpc.net/problem/15650

15650번: N과 M (2)

 

 

 

난 백트래킹에 정말 약한 것 같다!!

이것도 오늘 몇시간동안 찾아보고 고민한 것.. ㅜㅜ 흑 현타온다.

 

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

---

 

지난 포스팅의 코드를 응용하려고 노력했다. 다른 점은 반복문에서 i 의 시작점을 지정해 주는 것이다.

중복없이 >>오름차순<<으로 골라야 하니까 재귀함수를 썼을 때 시작점은 현재 위치의 다음 칸이다.

N,M = map(int, input().split())# 입력 받기

check = [False] * N # 사용했는지 안했는지 check 해 줄 배열, 길이는 숫자 개수 N과 같다.
answer = [0] * M # 정답 배열은 길이가 M 이다.

def dfs(idx, start):
    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+1) # 정답 배열의 다음 원소를 찾으러 재귀함수를 사용한다.
                check[i] = False # 중요!) 다 사용했으면 False로 바꿔 사용하지 않았다고 표시 해 준다.

dfs(0,0)