[JAVA] 백준 15652 - N과 M (4)

 

실버III

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

15652번: N과 M (4)

---

📌 문제

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

• 1부터 N까지 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 고른 수열은 비내림차순이어야 한다.
  • 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

---

📤 입력

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

---

📥 출력

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

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

---

💡 접근방법

중복된 수열을 출력하지 않으려면 출력되는 arr요소의 다음 요소가 더 커야 한다는 규칙을 찾았다.

따라서 출력할 때, 출력되는 arr요소의 다음 요소가 더 작으면 check에 true를 주어 중복되어 출력되었다는 것을 판별하였다.

---

💻 코드

package 백트래킹;

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

public class SLIVER3_BJ15652 {
    /*https://www.acmicpc.net/problem/15652
    N과M(4)
     */

    public static int[] arr;
    public static boolean[] visit;

    public static void main(String args[]) throws IOException{

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        //N과 M입력
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        arr = new int[M];
        visit = new boolean[N];

        dfs(N, M, 0);
        bw.flush();
    }

    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    public static void dfs(int n, int m, int depth) throws IOException {

        int temp = 0;

        if(m == depth){
            boolean check = false;

            for(int var: arr) {
                //arr요소가 temp보다 작으면 check를 true로 변경한다.
                //(이미 출력했다는 표시)
                if(temp > var) {
                    check = true;
                    break;
                }
                //크면 temp를 현재 arr요소의 다음 요소로 변경한다.
                temp = var;
            }

            //check가 false면 출력
            if(check  == false) {
                for (int var : arr) {
                    bw.write(var + " ");
                }
                bw.write("\n");
            }
        }

        else{
            for(int i = 0; i < n; i++){
                if(visit[i] == false){
                    arr[depth] = i + 1;
                    dfs(n, m, depth + 1);
                }
            }
        }
    }
}