백준 15655 N과 M (6)

백준 15655 N과 M (6)

1. 문제 링크

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

15655번: N과 M (6)

 

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

1. Input을 배열로 받아 정렬하는 로직이 추가되었다.
2. 이후에는 일반적인 조합 로직을 적용하여, 중복값을 피하기 위해 visit을 체크한다.
3. Input에 따라 Output 양이 많아져서, StringBuilder를 사용하여 시간을 줄일 수 있다.

 

3. 코드

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

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];

        String[] line = br.readLine().split(" ");
        for (int i = 0; i < N; i++) {
            nums[i] = Integer.parseInt(line[i]);
        }

        Arrays.sort(nums);

        combination(0, 0);

        System.out.print(sb);
    }

    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. 조합 로직 외에 Input에 대한 정렬이 필요했다.
2. Output이 많은 경우, StringBuilder를 적극 활용하자