백준 15649 N과 M (1)

1. 문제 링크

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

2. 풀이

1. 순열은 구성 값이 동일해도 순서가 다르면 독립으로 카운트한다.
2. 따라서 재귀에서는 항상 nums의 0번째부터 시작하며 visit(중복 방지)을 체크해야한다.
3. 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 boolean[] visit;
    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];
        visit = new boolean[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;
        }
        /**
         * 순열은 순서가 다르면 다른 케이스로 취급한다.
         * 따라서 항상 0 부터 시작하여 방문하지 않은 수를 확인해야한다.
         */
        for (int i = 0; i < N; i++) {
            if (!visit[i]) {
                visit[i] = true;
                candidate[depth] = nums[i];
                permutation(depth + 1);
                visit[i] = false;

            }
        }
    }
}

4. 채점 결과

5. 느낀 점

1. 순열의 전형적인 문제
2. 재귀를 들어갔다 나오면서 visit 배열을 원래 상태로 만들어야한다.
3. Output이 많은 경우, StringBuilder를 적극 활용하자