UVA 10304 dp的四边形不等式优化

  最优二叉搜索树问题，设状态dp(l,r)为l到r构成一颗二叉树的最小花费，则有状态转移方程dp(l,r)=min{dp(l,k-1)+dp(k+1,r)+w(l,r)} (l<=k<=r)。w(l,r)的计算可以用前缀和得到，设sum(i)为前i个数的和，e(i)为第i个元素，则w(l,r)=sum(r)-sum(l-1)-e(k)。

  可以用记忆化搜索求解，需要注意边界条件即l>r是返回0，由于函数调用所消耗的时间，记忆化搜索这题用了4.725s。

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int e[260],sum[260]={0},dp[260][260];
int dfs(int l,int r)
{
	if(l>r) return 0;
	int& t=dp[l][r];
	if(t!=-1) return t;
	t=0x7FFFFFFF;
	for(int rt=dp[l][r-1];rt<=dp[l+1][r];rt++){
		t=min(t,dfs(l,rt-1)+dfs(rt+1,r)+sum[r]-sum[l-1]-e[rt]);
	}
	return t;
}
int main()
{
	int n;
	while(cin>>n){
		memset(dp,-1,sizeof(dp));
		for(int i=1;i<=n;i++) {cin>>e[i];sum[i]=sum[i-1]+e[i];}
		cout<<dfs(1,n)<<endl;
	}
	return 0;
}

  将记忆化搜索改写为递推式，则枚举区间长度，在枚举左端点，解题时间减少到了1.445s。

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int e[260],sum[260]={0},dp[260][260];
int main()
{
	int n;
	while(cin>>n){
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++) {cin>>e[i];sum[i]=sum[i-1]+e[i];}
		int l,r;
		for(int len=2;len<=n;len++)
			for(l=1;(r=l+len-1)<=n;l++){
				int& t=dp[l][r];
				t=0x7FFFFFFF;
				for(int k=s[l][r-1];k<=s[l+1][r];k++)
					t=min(t,dp[l][k-1]+dp[k+1][r]+sum[r]-sum[l-1]-e[k]);
			}
		cout<<dp[1][n]<<endl;
	}
	return 0;
}

  这个状态方程可以用四边形不等式优化，可知对于i<i'<j<j'，w(i,j)+w(i',j')=w(i',j)+w(i,j')，所以可知满足四边形不等式，优化后时间为0.226s

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int e[260],sum[260]={0},dp[260][260];
int s[260][260];
int main()
{
	int n;
	while(cin>>n){
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=n;i++) {cin>>e[i];sum[i]=sum[i-1]+e[i];}
		int l,r;
		for(l=1;l<=n;l++) s[l][l]=l;
		for(int len=2;len<=n;len++)
			for(l=1;(r=l+len-1)<=n;l++){
				int &t=dp[l][r];
				t=0x7FFFFFFF;
				for(int k=s[l][r-1];k<=s[l+1][r];k++){
					int temp=dp[l][k-1]+dp[k+1][r]+sum[r]-sum[l-1]-e[k];
					if(t>=temp) {t=temp;s[l][r]=k;}
				}
			}
		cout<<dp[1][n]<<endl;
	}
	return 0;
}