HDU-2829 Lawrence（dp+四边形优化）

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=2829

#include<cstdio>
#include<iostream>
#include<cstring>
#define N 1005
#define inf 0x3f3f3f3f
#include<algorithm>
#define ll long long
using namespace std;
ll a[N],dp[N][N],s[N][N],cost[N][N],val[N][N];
int main() {
	int n,m;
	while(~scanf("%d%d",&n,&m)&&(n||m)) {
		for(int i=1; i<=n; i++)
			scanf("%lld",&a[i]);
			
		memset(cost,0,sizeof(cost));
		memset(val,0,sizeof(val));
		
		for(int i=1; i<n; i++)//单个积的和 
			for(int j=i+1; j<=n; j++)
				val[i][j]=val[i][j-1]+a[i]*a[j];
				
		for(int i=n-1; i>=1; i--)//所有积的和 
			for(int j=i+1; j<=n; j++)
				cost[i][j]=cost[i+1][j]+val[i][j];
				
		for(int i=0; i<=m; i++)
			for(int j=0; j<=n; j++)
				dp[i][j]=inf;
				
		for(int i=1; i<n; i++) { //dp[i][j]表示j个节点上有i个炸弹
			dp[0][i]=cost[1][i];
			s[0][i]=1;//s记录区间最优解的下标
		}
		
		for(int i=1; i<=m; i++) {//四边形优化 
			s[i][n+1]=n-1;
			for(int j=n; j>i; j--)
				for(int k=s[i-1][j]; k<=s[i][j+1]; k++)
					if(dp[i][j]>dp[i-1][k]+cost[k+1][j]) {
						dp[i][j]=dp[i-1][k]+cost[k+1][j];
						s[i][j]=k;
					}
		}
		printf("%lld\n",dp[m][n]);
	}
}