LintCode "Copy Books"

class Solution {
public:
    /**
     * @param pages: a vector of integers
     * @param k: an integer
     * @return: an integer
     */
    int copyBooks(vector<int> &pages, int k) {
        size_t n = pages.size();
        if(k > n)
        {
            return *max_element(pages.begin(), pages.end());
        }

        //    Prefix Sums
        vector<long long> psum(n);
        for(int i = 0; i < n; i ++)
            psum[i] = i == 0? pages[i] : (psum[i - 1] + pages[i]);

        //  DP
        vector<vector<long long>> dp(n + 1, vector<long long>(k + 1, INT_MAX));
        for(int i = 1; i <= n; i ++)
            dp[i][1] = psum[i - 1];

        for(int i = 2; i <= k; i ++) // person
        for(int b = i; b <= n; b ++) // book
        for(int c = i-1; c < b; c ++) // prev book
        {
            long long last = dp[c][i - 1];
            long long cur  = psum[b-1] - psum[c - 1];
            dp[b][i] = min(dp[b][i], max(cur, last));
        }

        return dp[n][k];
    }
};