본문 바로가기
알고리즘/백준 - 실버

백준 15655 N과 M (6)

by 호롤롤로루야 2022. 1. 3.

백준 15655 N과 M (6)

1. 문제 링크

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

 

15655번: N과 M (6)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net

 

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를 적극 활용하자

 

'알고리즘 > 백준 - 실버' 카테고리의 다른 글

백준 15657 N과 M (8)  (0) 2022.01.05
백준 15656 N과 M (7)  (0) 2022.01.04
백준 15654 N과 M (5)  (0) 2022.01.02
백준 15652 N과 M (4)  (0) 2022.01.01
백준 15651 N과 M (3)  (0) 2021.12.31

댓글