解题思路:
以下分析摘自:https://blog.csdn.net/xingyeyongheng/article/details/26009171
假定dp[i][j]表示前i个数分成j段的最小值
cost[i]表示从1~i的数两两相乘的总和
sum[i]表示前i个数的和
则:
dp[i][j]=Min(dp[k][j-1]+cost[i]-cost[k]-sum[k]*(sum[i]-sum[k]))
=>dp[i][j]=dp[k][j-1]+cost[i]-cost[k]-sum[i]*sum[k]+sum[k]*sum[k]
由于有sum[i]*sum[k]这一项,所以不可能用单调队列维护
-cost[k]-sum[i]*sum[k]+sum[k]*sum[k]的最小值
所以我们要把sum[i]独立出来以便求维护单调队列是和i无关
现在我们需要找出最优的k,
令k2<k时k时最优的,即前k个数k为最优的取值
所以满足:
dp[k][j-1]+cost[i]-cost[k]-sum[i]*sum[k]+sum[k]*sum[k]
<=dp[k2][j-1]+cost[i]-cost[k2]-sum[i]*sum[k2]+sum[k2]*sum[k2]
=>(dp[k][j-1]-cost[k]+sum[k]*sum[k]-(dp[k2][j-1]-cost[k2]+sum[k2]*sum[k2]))/(sum[k]-sum[k2])<=sum[i]
设:
y1=dp[k][j-1]-cost[k]+sum[k]*sum[k]
x1=sum[k]
y2=dp[k2][j-1]-cost[k2]+sum[k2]*sum[k2]
x2=sum[k2]
所以变成:
(y2-y1)/(x2-x1)
即两点之间的斜率!
要注意斜率dp的形式!!!sum[i]必须是单调递增才行。并且y和x中不含有i,且有能化出一个形式(y2-y1)/(x2-x1)<sum[i].然后直接套模板吧 。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1005
using namespace std;
int sum[N],cost[N],s[N],dp[N][N],stk[N];
int n,m;
int getup(int k1,int k2,int j)
{
return dp[k1][j]-cost[k1]+sum[k1]*sum[k1]-(dp[k2][j]-cost[k2]+sum[k2]*sum[k2]);
}
int getdown(int k1,int k2)
{
return sum[k1]-sum[k2];
}
int getdp(int k,int i,int j)
{
return dp[k][j]+cost[i]-cost[k]-sum[i]*sum[k]+sum[k]*sum[k];
}
void solve()
{
for(int i=1;i<=n;i++) dp[i][0]=cost[i];
for(int j=1;j<=m;j++)
{
int head=0,tail=0;
stk[tail++]=0;
for(int i=1;i<=n;i++)
{
while(head+1<tail&&getup(stk[head+1],stk[head],j-1)<sum[i]*getdown(stk[head+1],stk[head]))
head++;
dp[i][j]=getdp(stk[head],i,j-1);
while(head+1<tail&&getup(stk[tail-1],stk[tail-2],j-1)*getdown(i,stk[tail-1])>getup(i,stk[tail-1],j-1)*getdown(stk[tail-1],stk[tail-2]))
tail--;
stk[tail++] =i;
}
}
}
int main()
{
//freopen("t.txt","r",stdin);
while(scanf("%d%d",&n,&m)!=EOF)
{
if(n+m==0) break;
for(int i=1;i<=n;i++)scanf("%d",&s[i]);
for(int i=1;i<=n;i++)sum[i]=sum[i-1]+s[i];
memset(cost,0,sizeof cost);
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n;j++)cost[j]+=s[i]*s[j];
}
for(int i=1;i<=n;i++)cost[i]+=cost[i-1];
solve();
printf("%d\n",dp[n][m]);
}
return 0;
}