[백준/자바] 15657번: N과 M (8)

문제

https://www.acmicpc.net/problem/15657

풀이

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 오름차순

중복 조합을 찾는 문제

15652번: N과 M (4)은 1부터 N까지의 중복 조합을 찾는 문제였지만, 이 문제는 입력 받은 자연수들의 중복 조합을 찾는 문제입니다.

		st = new StringTokenizer(br.readLine());
		nums = new int[n];
		for (int i = 0; i < n; i++) nums[i] = Integer.parseInt(st.nextToken());
		Arrays.sort(nums);

• 오름차순으로 출력해야 하기 때문에, 입력 받을 때 정렬을 해줬습니다.

	private static void dfs(int idx, int start) { // 조합 배열 인덱스, 시작할 nums 배열 인덱스
		
		if (idx == m) { // m개를 골랐다면 출력
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		for (int i = start; i < n; i++) { // 시작 인덱스부터 시작하여 모든 값 삽입
			li[idx] = nums[i];
			dfs(idx + 1, i); // 재귀
		}
	}

• 중복된 값들도 넣어 줘야 하기 때문에, visited 배열은 필요하지 않습니다.

코드

import java.io.*;
import java.util.*;

public class Main_15657 {
	static StringBuilder sb = new StringBuilder();
	static int n, m;
	static int[] li, nums;
	
	private static void dfs(int idx, int start) {
		
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		for (int i = start; i < n; i++) {
			li[idx] = nums[i];
			dfs(idx + 1, i);
		}
	}
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		n = Integer.parseInt(st.nextToken());
		m = Integer.parseInt(st.nextToken());
		
		st = new StringTokenizer(br.readLine());
		nums = new int[n];
		for (int i = 0; i < n; i++) nums[i] = Integer.parseInt(st.nextToken());
		Arrays.sort(nums);
		
		li = new int[m];
		dfs(0, 0);
		
		System.out.println(sb.toString());
	}

}