백준 - 15651 N과 M (3) [백트래킹]

 

15651번: N과 M (3)

 

1. 문제 요약

1부터 N까지의 숫자들 중 M개를 뽑아 수열을 생성하라.

같은 수를 여러 번 골라도 된다.

 

2. 접근 방법

대표적인 방법으로 재귀함수를 이용할 수 있습니다.

 

3. 파이썬

from sys import stdin
input = stdin.readline

def solution(depth: int):
    if depth == M:
        # print(*arr)
        print(' '.join(map(str, arr))) # 위 방식보다 좀 더 효율적입니다.
        return

    for i in range(1, N + 1):
        arr[depth] = i
        solution(depth + 1)

N, M = map(int, input().split())
arr = [0] * M
solution(0)

 

4. 자바

static int N, M;
static int[] arr; // new int[M]
static StringBuilder sb = new StringBuilder();

static void solution(int depth) {
    if (depth == M) {
        for (int i : arr) {
            sb.append(i).append(' ');
        }
        sb.append('\n');
        return;
    }

    for (int i = 1; i <= N; i++) {
        arr[depth] = i;
        solution(depth + 1);
    }
}

 

5. 전체 코드

https://github.com/Ksiyeong/Algorithm/tree/main/%EB%B0%B1%EC%A4%80/Silver/15651.%E2%80%85N%EA%B3%BC%E2%80%85M%E2%80%85%EF%BC%883%EF%BC%89