백준 15650 N과 M (2)

1. 문제 링크

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

2. 풀이

1. 조합은 순서와 상관 없이 동일 구성이면 하나로 카운트한다.
2. 따라서 이전 재귀에 접근했던 값을 피하기 위해 다음 재귀로 넘어갈 시, i+1을 한고 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;
        }

        combination(0, 0);

        System.out.print(sb);
    }

    /**
     * 조합은 순서와 상관 없이 구성이 같으면 1개로 치부한다.
     * 따라서, for 문에서 이전 단계의 수에 접근하지 않도록 다음 재귀로 넘길 때 i + 1을 해준다
     */
    static void combination(int depth, int start) {
        if (depth == M) {
            for (int i = 0; i < M; i++) {
                sb.append(candidate[i]).append(" ");
            }
            sb.append("\n");
            return;
        }

        for (int i = start; i < N; i++) {
            if (!visit[i]) {
                visit[i] = true;
                candidate[depth] = nums[i];
                combination(depth + 1, i + 1);
                visit[i] = false;

            }
        }
    }
}

4. 채점 결과

5. 느낀 점

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