石子合并 (区间dp)

题目描述:

 在一个操场上摆放着一行共n堆的石子。现要将石子有序地合并成一堆。规定每次只能选相邻的两堆合并成新的一堆，并将新的一堆石子数记为该次合并的得分。请编辑计算出将n堆石子合并成一堆的最小得分和将n堆石子合并成一堆的最大得分。

Input:

 输入第一行为n(n<1000)，表示有n堆石子，第二行为n个用空格隔开的整数，依次表示这n堆石子的石子数量（<=1000）

Output:

 输出将n堆石子合并成一堆的最小得分和将n堆石子合并成一堆的最大得分

#include 
#include 
#include 
using namespace std;

#define MAX     0
#define MIN     1
#define STATUS  2
#define MAXN    1001
#define INF     0x7ffffff

int val[MAXN];
int dp[MAXN][MAXN][STATUS], delta[MAXN][MAXN];

int dfs(const int &l, const int &r, const int &mark)
{
        if( l == r ) {
                return 0;
        }
        if( -1 != dp[l][r][mark] ) {
                return dp[l][r][mark];
        }
        int rst, l_rst, r_rst;
        rst = (MAX == mark)? 0 : INF;
        for(int i = l; i < r; i ++) {
                l_rst = dfs(l, i, mark); r_rst = dfs(i+1, r, mark);
                if( MAX == mark ) {
                        rst = max(rst, l_rst+r_rst+delta[l][r]);
                }
                else {
                        rst = min(rst, l_rst+r_rst+delta[l][r]);
                }
        }
        return dp[l][r][mark] = rst;
}

int main(int argc, char const *argv[])
{
#ifndef ONLINE_JUDGE
        freopen("test.in", "r", stdin);
#endif
        int n;
        scanf("%d", &n);
                for(int i = 0; i < n; i ++) {
                        scanf("%d", &val[i]);
                }
                memset(dp, -1, sizeof(dp));
                for(int i = 0; i < n; i ++) {
                        for(int j = i, sum = 0; j < n; j ++, sum = 0) {
                                for(int k = i; k <= j; k ++) {
                                        sum += val[k];
                                }
                                delta[i][j] = sum;
                        }
                }
                printf("%d %d\n", dfs(0, n-1, MIN), dfs(0, n-1, MAX));
        return 0;
}