Lawrence（hdu 2829）

题目：hdu 2829

题意：给定一个长度为n的序列，至多将序列分成m+1段，每段序列都有权值，权值为序列内两个数两两相乘之和。求最小序列权值和。

思路：

dp[i][j]：把前 i 个数分成 j 段后能得到的序列的最小权值和 ;

cost[i]：前 i 个数的序列权值和 ;

sum[i] : 前缀和 ;

状态转移方程： dp[i][j] = min( dp[k][j-1] + cost[i] - cost[k] - sum[k] *(sum[i] - sum[k]) ) ;

=> sum[i]*sum[k] + dp[i][j] - cost[i] = dp[k][j-1] - cost[k] + sum[k]*sum[k] ;

=>设 x: sum[k] , b: dp[i][j] - cost[i] , y: dp[k][j-1] - cost[k] + sum[k]*sum[k] ;

=>斜率k = sum[i] ;

=>斜率计算: k = (y1-y2) / (x1-x2) ;

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

const int MAX = 1000+10;

long long a[MAX];
long long sum[MAX],cost[MAX];
long long q[MAX];
long long dp[MAX][MAX];

long long up(int j,int k1,int k2)
{
    return ((dp[k1][j-1]-cost[k1]+sum[k1]*sum[k1])-(dp[k2][j-1]-cost[k2]+sum[k2]*sum[k2]));
}

long long down(int k1,int k2)
{
    return (sum[k1]-sum[k2]);
}

int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m),n||m)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        sum[0]=0;
        cost[0]=0;
        for(int i=1;i<=n;i++)
        {
            sum[i]=sum[i-1]+a[i];
            cost[i]=cost[i-1]+sum[i-1]*a[i];
        }
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
        {
            dp[i][0]=cost[i];
            dp[i][i-1]=0;
        }
        for(int j=1;j<=m;j++)
        {
            int Left=1;
            int Right=1;
            q[1]=j;
            for(int i=j+1;i<=n;i++)
            {
                while(Left<Right&&up(j,q[Left+1],q[Left])<=sum[i]*down(q[Left+1],q[Left]))
                    Left++;
                int Front=q[Left];
                dp[i][j]=dp[Front][j-1]+cost[i]-cost[Front]-sum[Front]*(sum[i]-sum[Front]);
                while(Left<Right&&up(j,q[Right],q[Right-1])*down(i,q[Right])>=up(j,i,q[Right])*down(q[Right],q[Right-1]))
                    Right--;
                q[++Right]=i;
            }
        }
        printf("%lld\n",dp[n][m]);
    }
    return 0;
}