백준 15657 N과 M (8)

백준 15657 N과 M (8)

1. 문제 링크

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

15657번: N과 M (8)

 

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

        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++) {
            candidate[depth] = nums[i];
            combination(depth + 1, i);
        }
    }
}

4. 채점 결과

5. 느낀 점

1. 중복 조합 로직 외에 Input에 대한 정렬이 필요했다.
2. Output이 많은 경우, StringBuilder를 적극 활용하자