백준 15652: N과 M (4) [Java] - 포포
2022. 6. 4. 17:04ㆍ알고리즘
문제
자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.
- 1부터 N까지 자연수 중에서 M개를 고른 수열
- 같은 수를 여러 번 골라도 된다.
- 고른 수열은 비내림차순이어야 한다.
- 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.
입력
첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
출력
한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.
수열은 사전 순으로 증가하는 순서로 출력해야 한다.
내 제출
package com.company.ch11BackTracking;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
static int n, size;
static int[] arr;
static StringBuffer sb;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
sb = new StringBuffer();
n = Integer.parseInt(st.nextToken(" "));
size = Integer.parseInt(st.nextToken(" "));
arr = new int[size];
used = new boolean[n+1];
solve(0, 1);
System.out.println(sb);
}
private static void solve(int depth, int start) {
if(depth == size){
for (int i = 0; i < size; i++) {
sb.append(arr[i]+" ");
}
sb.append('\n');
return;
}
for (int i = start; i <= n; i++) {
arr[depth] = i;
solve(depth+1, i);
}
}
}
대표적인 백트래킹 문제다.
N과 M 시리즈 4문제를 풀어보니 백트래킹 좀 할만한가..? 라는 착각이 든다.
재귀는 아직도 어렵다. 파이팅~
'알고리즘' 카테고리의 다른 글
프로그래머스: 더 맵게(lv2) [Java] (0) | 2022.08.30 |
---|---|
프로그래머스: 뉴스 클러스터링(lv2) [Java] (0) | 2022.08.17 |
프로그래머스: 카카오프렌즈 컬러링북(lv2) (0) | 2022.08.12 |
프로그래머스 2차 문제집(모의고사) 후기 (0) | 2022.08.10 |
Union find(합집합 찾기, 서로소 집합) 알고리즘 (0) | 2022.05.24 |