16-janghw0126

[janghw0126]

🔗 문제 링크

백준 | N과 M (5)

문제 설명

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

제한 조건

• N개의 자연수는 모두 다른 수이다.
• 수열은 N개의 자연수 중에서 M개를 고른 수열이다.

입력

• 첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
• 둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

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

✔️ 소요된 시간

1시간

✨ 수도 코드

✅이 문제의 아이디어는 주어진 숫자들을 배열로 받은 뒤 정렬하고, 수열로 만든다는 것이다. 그래서 백트래킹 알고리즘의 유형에 맞게, 모든 경우의 수열을 재귀적으로 탐색하여 출력하였다. 이때, 사전 순서대로 결과값을 출력하는 것이기 때문에 DFS 방식으로 경우를 탐색하는 것이 더 적절한 것 같았다.

📖풀이 방법

1. 자연수 N과 M을 입력받는다.
2. 숫자들을 입력받아 리스트에 저장한다.
3. 숫자들을 오름차순으로 정렬한다.
4. 재귀 함수로 사용할 backtracking(depth) 함수를 선언한다.
5. 현재까지 선택한 숫자의 개수가 M과 같다면 박스에 저장된 숫자들을 문자열로 변환하여 공백을 기준으로 결합하여 출력한다.
6. 0부터 N-1까지의 인덱스에 대해 반복한다.
7. 이미 선택된 숫자라면 건너뛴다.
8. 숫자를 박스에 추가하고 다음 자릿수의 숫자를 선택하기 위해 재귀적으로 backtracking 함수를 호출한다.
9. 함수 호출 후에는 해당 자릿수에서의 선택이 끝났으므로, 박스에서 마지막에 추가된 숫자를 제거한다.
10. 함수를 빠져나와 선택한 숫자들을 저장할 리스트를 선언한다.
11. 백트래킹 함수를 호출하여 문제를 해결한다.

⬇️전체 코드

#자연수 N과 M을 입력받는다.
N, M = map(int, input().split())
#숫자들을 입력받아 리스트에 저장한다.
numbers = [int(x) for x in input().split()]
#숫자들을 오름차순으로 정렬한다.
numbers.sort()

def backtracking(depth):
    #현재까지 선택한 숫자의 개수가 M과 같다면 수행할 문이다.
    if depth == M:
        #박스에 저장된 숫자들을 문자열로 변환하여 공백을 기준으로 결합하여 출력한다.
        print(' '.join(map(str,box)))
        return
    #0부터 N-1까지의 인덱스에 대해 반복한다.
    for i in range(N):
        #이미 선택된 숫자라면 건너뛴다.
        if numbers[i] in box:
            continue
        #숫자를 박스에 추가한다.
        box.append(numbers[i])
        #다음 자릿수의 숫자를 선택하기 위해 재귀적으로 backtracking 함수를 호출한다.
        backtracking(depth + 1)
        #함수 호출 후에는 해당 자릿수에서의 선택이 끝났으므로, 박스에서 마지막에 추가된 숫자를 제거한다.
        box.pop()
#선택한 숫자들을 저장할 리스트를 선언한다.
box = []
#백트래킹 함수를 호출하여 문제를 해결한다.
backtracking(0)

📚 새롭게 알게된 내용

.

[LJEDD2]

저도 같은 방법으로 풀었었네요 ㅎㅎ

input = open(0).readline
n,m = map(int,input().split())
nlist = sorted(list(map(int,input().split())))
result = []

def dfs(depth):

    if len(result) == m:
        return print(*result)

    for i in nlist:
        if i not in result:
            result.append(i)
            dfs(depth + 1)
            result.pop()
dfs(0)

이렇게 리스트 요소에 바로 접근해서 숫자를 저장하고 출력을 할 수도 있습니다 !

여행 준비로 바쁘셨을텐데 정성스러운 PR 남겨주시느라 고생하셨습니다 !! ☘️👍

[avocado-13]

혜원님과 정은님의 코드를 보며 다양하게 코드를 짤 수 있다는 것을 다시금 느낍니다🤓
이번 차시 PR 작성하시느라 수고 많으셨습니다 👍🏻🍀