[BOJ][1201] NMK

1. 문제 링크

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

2. 사용 알고리즘

그리디

3. 풀이

1) 배열에 1 ~ N을 차례대로 저장

2) 적어도 한 개의 그룹은 K개의 수를 갖도록 배열을 M개의 그룹으로 나눔

3) 각 그룹의 수를 내림차순 정렬

• 각 그룹별 첫 번째 수끼리 가장 긴 증가하는 부분수열의 길이가 M

• K개의 수를 갖는 그룹의 가장 긴 감소하는 부분수열의 길이가 K

조건1) M개의 그룹이 모두 K개의 수로 이루어진 경우 총 M * K개

∴ N <= M * K

조건2) M개의 가장 긴 증가하는 부분수열, K개의 가장 긴 감소하는 부분수열일 때 중간에 한 개의 수가 겹치므로 1을 빼주어야 한다.

∴ N >= M + K - 1

4. 소스 코드

4-1. C++

https://github.com/dev-aiden/problem-solving/blob/main/boj/1201.cpp

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> v(503);

int main(void) {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    int n, m, k; cin >> n >> m >> k;
    if(n >= m + k - 1 && n <= m * k) {
        for(int i = 0; i < n; ++i) v[i] = i + 1;
        int index = 0;
        for(int i = 1; i <= m; ++i) {
            int temp = min(index + k, n - m + i);
            reverse(v.begin() + index, v.begin() + temp);
            index = temp;
        }
        for(int i = 0; i < n; ++i) cout << v[i] << " ";
        cout << "\n";
    } else {
        cout << -1 << "\n";
    }
    return 0;
}

4-2. JAVA

https://github.com/dev-aiden/problem-solving/blob/main/boj/1201.java

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.StringTokenizer;

public class Main {

    static int[] arr = new int[503];
    static ArrayList<Integer> a = new ArrayList<>();

    public static void main(String[] args) throws IOException  {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());
        int k = Integer.parseInt(st.nextToken());
        if(n >= m + k - 1 && n <= m * k) {
            for(int i = 0; i < n; ++i) arr[i] = i  + 1;
            int index = 0;
            for(int i = 1; i <= m; ++i) {
                int temp = Math.min(index + k, n - m + i);
                for(int j = temp - 1; j >= index; --j) a.add(arr[j]);
                index = temp;
            }
            StringBuilder sb = new StringBuilder();
            for(int i = 0; i < n; ++i) sb.append(a.get(i)).append(" ");
            sb.append("\n");
            System.out.println(sb);
        } else {
            System.out.println(-1);
        }
    }
}