hdu 2829 Lawrence

http://acm.hdu.edu.cn/showproblem.php?pid=2829

最近在看斜率优化的dp，早都该学习了，一直拖到现在才学习……其实单调队列+斜率的DP只要推出X,Y,还有斜率之后,差不多就是模板题了.

这个题目我推的公式和网上大多不太一样，观察一下 （a+b+c）^2  - (a*a + b*b +c*c) = 2*(a*b + b*c + a*c) 那么 就可得 cost[i+1 ] [j ] = [  (sum[j] -sum[i])^2 - (sump[j]-sump[i])  ] /2 

然后dp[ k ][ i ] =min(dp[k-1][j ]  +  (sum[i ]-sum[j])^2 - (sump[i]-sump[j] ) )  化简为： dp[ k] [ i ]  =min(dp [k-1] [ j ] + sum[j ]^2 + sump[j]-2*sum[j]*sum[i]+ sum[i]^2 ,   然后

y:  dp [k-1][j] +sum[j]^2+sump[i]

x:  sum[j]

斜率： 2* sum[i] （当i确定时斜率就是常数，如果随着i的增大，sum[i] 变大）

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define inf 0x3fffffff
const int maxn=1010;
int dp[maxn][maxn],sum[maxn],sump[maxn],a[maxn],q[maxn];
int dy(int k,int j1,int j2)
{
    return dp[k][j1]+sum[j1]*sum[j1]+sump[j1]-(dp[k][j2]+sum[j2]*sum[j2]+sump[j2]);
}
int dx(int j1,int j2)
{
    return sum[j1]-sum[j2];
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m),(n||m))
    {
         for(int i=1;i<=n;i++) scanf("%d",&a[i]);
         for(int i=1;i<=n;i++)
         {
             sum[i]=sum[i-1]+a[i];
             sump[i]=sump[i-1]+a[i]*a[i];
         }
         for(int i=1;i<=n;i++) dp[0][i]=sum[i]*sum[i]-sump[i];
         int ans=inf;
         for(int k=1;k<=m;k++)
         {
             int l=0,r=0;
             for(int i=1;i<=n;i++)
             {
                 while(l+1<r&&dy(k-1,q[r-2],q[r-1])*dx(q[r-1],i-1)>=dy(k-1,q[r-1],i-1)*dx(q[r-2],q[r-1])) r--;
                 q[r++]=i-1;
                 while(l+1<r&&dy(k-1,q[l],q[l+1])>=2*dx(q[l],q[l+1])*sum[i]) l++;
                 dp[k][i]=dp[k-1][q[l]]+(sum[i]-sum[q[l]])*(sum[i]-sum[q[l]])-(sump[i]-sump[q[l]]);
             }
            ans=min(ans,dp[k][n]);
         }
        printf("%d\n",ans/2);
    }
    return 0;
}