개발일지/Algorithm
백준 - 15650 N과 M (6) [조합]
E-room
2023. 9. 28. 18:52
728x90
15655번: N과 M (6)
N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열
www.acmicpc.net
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. 전체 코드
728x90