代码改变世界

The Painter's Partition Problem Part I

2014-12-20 14:22  李涛的技术博客  阅读(824)  评论(0编辑  收藏  举报

(http://leetcode.com/2011/04/the-painters-partition-problem.html)

You have to paint N boards of lenght {A0, A1, A2 ... AN-1}. There are K painters available and you are also given how much time a painter takes to paint 1 unit of board. You have to get this job done as soon as possible under the constraints that any painter will only paint continues sections of board, say board {2, 3, 4} or only board {1} or nothing but not board {2, 4, 5}.

We define M[n, k] as the optimum cost of a partition arrangement with n total blocks from the first block and k patitions, so

                 n              n-1
M[n, k] = min { max { M[j, k-1], Ai } } j=1 i=j

The base cases are:

M[1, k] = A0
         n-1
M[n, 1] = Σ Ai
i=0

Therefore, the brute force solution is:

int sum(int A[], int from, int to)
{
    int total = 0;
    for (int i = from; i <= to; i++)
        total += A[i];
    return total;
}

int partition(int A[], int n, int k)
{
    if (n <= 0 || k <= 0)
        return -1;
    if (n == 1)
        return A[0];
    if (k == 1)
        return sum(A, 0, n-1);

    int best = INT_MAX;
    for (int j = 1; j <= n; j++)
        best = min(best, max(partition(A, j, k-1), sum(A, j, n-1)));

    return best;
}

 It is exponential in run time complexity due to re-computation of the same values over and over again.

The DP solution:

int findMax(int A[], int n, int k)
{
    int M[n+1][k+1];
    int sum[n+1];
    for (int i = 1; i <= n; i++)
        sum[i] = sum[i-1] + A[i-1];

    for (int i = 1; i <= n; i++)
        M[i][1] = sum[i];
    for (int i = 1; i <= k; i++)
        M[1][k] = A[0];

    for (int i = 2; i <= k; i++)
    {
        for (int j = 2; j <= n; j++)
        {
            int best = INT_MAX;
            for (int p = 1; p <= j; p++)
            {
                best = min(best, max(M[p][i-1], sum[j]-sum[p]));
            }
            M[j][i] = best;
        }
    }
    return M[n][k];
}

Run time: O(kN*N), space complexity: O(kN).