백준 15651 N과 M (3)

백준 15651 N과 M (3)

1. 문제 링크

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

15651번: N과 M (3)

2. 문제 해결에 대한 아이디어

1. 중복 순열이므로 순열 로직에서 visit을 제외하였다.
2. Input에 따라 Output 양이 많아져서, StringBuilder를 사용하여 시간을 줄일 수 있다.

 

3. 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {

    static int N, M;
    static int[] nums;
    static int[] candidate;
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] nm = br.readLine().split(" ");
        N = Integer.parseInt(nm[0]);
        M = Integer.parseInt(nm[1]);

        nums = new int[N];
        candidate = new int[M];

        for (int i = 0; i < N; i++) {
            nums[i] = i + 1;
        }

        permutation(0);

        System.out.print(sb);
    }

    static void permutation(int depth) {
        if (depth == M) {
            for (int i = 0; i < M; i++) {
                sb.append(candidate[i]).append(" ");
            }
            sb.append("\n");
            return;
        }
        /**
         * 일반 순열과 달리 중복을 허용한다.
         * 따라서 visit을 따로 체크하지 않는다.
         */
        for (int i = 0; i < N; i++) {
            candidate[depth] = nums[i];
            permutation(depth + 1);
        }
    }
}

4. 채점 결과

5. 느낀 점

1. 중복 순열은 일반 순열과 달리 visit을 사용하지 않는다.
2. Output이 많은 경우, StringBuilder를 적극 활용하자