[BOJ] 11066 : 파일 합치기

문제 링크

11066번: 파일 합치기 (acmicpc.net)

11066번: 파일 합치기

풀이

모든 파일을 합치는 최소 비용을 출력하는 문제.

동적계획법(Dynamic programming)을 활용해야한다. 그 중에서도 2차원 배열을 이용한 DP를 활용해야한다.

DP[i][j] 배열에 i부터 j번째 까지의 파일을 합치기 위해 필요한 최소비용을 저장한다.

즉 DP[i][j] = Min(i <= K <= j)DP[i][K] + DP[K+1] + Sum[i to j] (이건 고정지출)

예제를 보면 40 30 30 50 이다 DP[i][i] 는 모두 0이 저장되어 있고

DP[1][2] = 70, DP[2][3] = 60, DP[3][4] = 50 이 할당된다.

DP[1][3]에는 DP[1][1] + DP[2][3] , DP[1][2] + DP[3][3] 중 최소값과 Sum[1 ~ 3]을 더한 값이 저장되어야한다.

 

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include <string.h>
#include <string>
#include <stack>
#include <vector>
#include <math.h>
#include <queue>
#include <climits>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef priority_queue <pair<pair<int, int>, int>, vector<pair<pair<int, int>, int>>, greater<pair<pair<int, int>, int>>> Priority_queue;
const int INF = INT_MAX;
const int MAX_N = 500 + 5;
int K;
int T;
int dp[MAX_N][MAX_N];
int File[MAX_N];
ll Sum[MAX_N];
void Init() {
    for (int i = 0; i < MAX_N; i++) {
        for (int j = 0; j < MAX_N; j++) {
            dp[i][j] = 0;
        }
    }
    memset(File, 0, sizeof(File));
    memset(Sum, 0, sizeof(Sum));
}
int main(void)
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> T;
    while (T--) {
        cin >> K;
        Init();
        for (int i = 1; i <= K; i++) {
            cin >> File[i];
            Sum[i] += Sum[i - 1] + File[i];
        }
        for (int i = 1; i < K; i++) { // 범위의 길이
            for (int j = 1; j <= K - i; j++) { //시작점
                int Min = INF;
                for (int k = 0; k < i; k++) {
                    Min = min(Min, dp[j][j + k] + dp[j+k+1][j + i]);
                }
                dp[j][j + i] =  Min + (Sum[j + i] - Sum[j - 1]);
            }
        }
        cout << dp[1][K] << "\n";
    }
    return 0;
}

구현만 잘 해준다면 AC를 받을 수 있다.