백준 - 15650 N과 M (6) [조합]

 

15655번: N과 M (6)

 

1. 문제 요약

주어진 숫자들 중 M개를 뽑아 조합(Combination)을 만드는 문제입니다.

고른 수열은 오름차순이어야 합니다.

 

2. 접근 방법

• 입력받은 숫자들을 오름차순 정렬
• 재귀함수를 이용하여 조합 생성

 

3. 파이썬

from sys import stdin
input = stdin.readline

def solution(start: int, depth: int):
    if depth == M:
        print(' '.join(map(str, arr)))
        return

    for i in range(start, N):
        if not visit[i]:
            visit[i] = True
            arr[depth] = nums[i]
            solution(i + 1, depth + 1)
            visit[i] = False

N, M = map(int, input().split())
nums = list(map(int, input().split()))
nums.sort()
visit = [False] * N
arr = [0] * M

solution(0, 0)

 

4. 자바

static int N, M;
static int[] nums; // 입력받은 숫자, 오름차순 정렬 해야함
static int[] arr;
static boolean[] visit;
static StringBuilder sb = new StringBuilder();

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

    for (int i = start; i < N; i++) {
        if (!visit[i]) {
            visit[i] = true;
            arr[depth] = nums[i];
            solution(i + 1, depth + 1);
            visit[i] = false;
        }
    }
}

 

5. 전체 코드

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