DP_石子合并问题

/*动态规划石子合并问题
*【石子合并】
*在一个圆形操场的四周摆放着n堆石子。现要将石子有次序地合并成一堆。规定每次只能选相邻的2堆石子合并**成新的一堆，并将新的一堆石子数记为该次合并的得分。
*试设计一个算法，计算出将n堆石子合并成一堆的最小得分和最大得分。
*【输入文件】包含两行，第1行是正整数n（1<=n<=100），表示有n堆石子。第2行有n个数，分别表示每堆石子**的个数。
*【输出文件】输出两行。第1行中的数是最小得分；第2行中的数是最大得分。
*【输入样例】
* 4
* 4 4 5 9
*【输出样例】
* 43
* 54

* */

核心在于状态转移方程：

求最大值：dp[i][j] = max{large(i, x) + large(x+1, j)} + sum(i, j);

求最小值：dp[i][j] = min{small(i, x) + small(x+1, j)} + sum(i, j);

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>

#define INPUT freopen("D:\\iotest\\in.txt", "r", stdin)
#define OUTPUT freopen("D:\\iotest\\out.txt", "a", stdout)
#define TIME printf("%.3lf s", (double)clock() / CLOCKS_PER_SEC)
#define TEST printf("this is an output test!!!")
using namespace std;

int dp[1000][1000];
int a[1000] = {0};

int sum(int i, int j)
{
	int s = 0;
	for(int k = i-1; k <= j-1; k++)
	    s += a[k];
	return s;
}

int large_value(int i, int j)
{
	int u = 0;
	if(dp[i][j] > 0) return dp[i][j];
	else if(i == j) return dp[i][j] = 0;
	else{
		dp[i][j] = large_value(i, i) + large_value(i+1, j) + sum(i, j);
		u = dp[i][j];
		for(int x = i+1; x < j; ++x)
		{ 
			dp[i][j] = large_value(i, x) + large_value(x+1, j) + sum(i, j);
			if(u > dp[i][j]) dp[i][j] = u;
			else u = dp[i][j];
		}
		return dp[i][j];
	}
}

int small_value(int i, int j)
{
	int u = 0;
	if(dp[i][j] > 0) return dp[i][j];
	else if(i == j) return dp[i][j] = 0;
	else{
		dp[i][j] = small_value(i, i) + small_value(i+1, j) + sum(i, j);
		u = dp[i][j];
		for(int x = i+1; x < j; ++x)
		{ 
			dp[i][j] = small_value(i, x) + small_value(x+1, j) + sum(i, j);
			if(u < dp[i][j]) dp[i][j] = u;
			else u = dp[i][j];
		}
		return dp[i][j];
	}
}

int main()
{
	INPUT;
	OUTPUT;
	int n;
	cin >> n;
	for(int i = 0; i < n; ++i)  cin >> a[i];
	memset(dp, -1, sizeof(dp));
	cout << large_value(1, n) << endl;
	memset(dp, -1, sizeof(dp));
	cout << small_value(1, n) << endl;
	return 0;
}