백준 BOJ 15649 - N과 M(1) : JAVA 자바 풀이

 

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

예제 입력 1

3 1

예제 출력 1

1
2
3

예제 입력 2

4 2

예제 출력 2

1 2
1 3
1 4
2 1
2 3
2 4
3 1
3 2
3 4
4 1
4 2
4 3

예제 입력 3

4 4

예제 출력 3

1 2 3 4
1 2 4 3
1 3 2 4
1 3 4 2
1 4 2 3
1 4 3 2
2 1 3 4
2 1 4 3
2 3 1 4
2 3 4 1
2 4 1 3
2 4 3 1
3 1 2 4
3 1 4 2
3 2 1 4
3 2 4 1
3 4 1 2
3 4 2 1
4 1 2 3
4 1 3 2
4 2 1 3
4 2 3 1
4 3 1 2
4 3 2 1

---

위 문제는 N개의 수를 중복 없이 M개의 수를 순서대로 나열 하는 것이다.

 

시간 복잡도

O(N! / (N - M)!) -> 8! / 0! = 40,320

공간 복잡도

O(M) -> M개의 수를 고르는 것이기에

 

import java.util.*;

public class no15649{
	static StringBuilder sb = new StringBuilder();
    static int N, M;
    static int[] selected;
    
    static void input(){
    	Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        M = sc.nextInt();
        selected = new int[M + 1];
    }
    
    static void rec_func(int k){
    	// k와 선택한 갯수의 수가 동일하면 수열을 출력하는 코드이다.
    	if(k == M + 1){
        	for(int i = 1; i <= M; i++){
            	sb.append(selected[i]).append(' ');
            }
            sb.append('\n');
        }else{
        // 만약 수가 동일하지 않다면 올바른 수를 저장한다.
        	for(int cand = 1; cand <= N; cand++){
            	// isUsed 불리언을 사용하는 이유는 중복 여부를 확인하기 위해서다.
            	boolean isUsed = false;
                for(int i = 1; i < k; i++){
                	// 만약 가지려는 값이 앞서 나왔다면 중복 되었다는 것을 확인한다.
                	if(cand == selected[i]) isUsed = true;
                }
                if(!isUsed){
                	selected[k] = cand;
                    rec_func(k + 1);
                    selected[k] = 0;
                }
            }
        }
    }
    
    public static void main(String[] args){
    	input();
        rec_func(1);
        System.out.println(sb.toString);
    }
}