[백준/자바] 15651번: N과 M (3)

문제

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

풀이

• 자연수 N과 M
• 1부터 N까지 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 고를 수 있다.

중복 순열을 구하는 문제

구할 수 있는 모든 경우를 출력하면 됩니다.

	private static void dfs(int idx) {
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		for (int i = 1; i <= n; i++) {
			li[idx] = i; // i번째 수 삽입
			dfs(idx + 1); // 재귀
		}
	}

재귀를 통해 모든 수를 넣어 줍니다.

• 방문 처리 역시 필요하지 않습니다.

코드

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

public class Main {
	static StringBuilder sb = new StringBuilder();
	static int n, m;
	static int[] li;
	
	private static void dfs(int idx) {
		if (idx == m) {
			for (int i : li) sb.append(i).append(" ");
			sb.append("\n");
			return;
		}
		
		for (int i = 1; i <= n; i++) {
			li[idx] = i;
			dfs(idx + 1);
		}
	}
	
    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());
    	
    	li = new int[m];
    	
    	dfs(0);
    	System.out.println(sb.toString());
    }
}