hdu 3480 Division

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

这个题目的数据范围确实有点大，1000*5000，即使n*m的时间复查度也让人有点不敢做

思路： 排序，斜率优化即可

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn=10010;
const int maxm=5010;
int dp[maxm][maxn],a[maxn],q[maxn];

int dy(int k,int j1,int j2)
{
    return dp[k][j2]+a[j2+1]*a[j2+1]-(dp[k][j1]+a[j1+1]*a[j1+1]);
}
int dx(int j1,int j2)
{
    return a[j2+1]-a[j1+1];
}
int main()
{
    int n,m,T,ca=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        if(m>=n){
           printf("Case %d: 0\n",ca++);
           continue;
        }
        sort(a+1,a+1+n);
        for(int i=1;i<=n;i++) dp[1][i]=(a[i]-a[1])*(a[i]-a[1]);
        for(int k=2;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])<=dx(q[l],q[l+1])*2*a[i]) l++;
                dp[k][i]=dp[k-1][q[l]]+(a[i]-a[q[l]+1])*(a[i]-a[q[l]+1]);
            }
        }
        printf("Case %d: %d\n",ca++,dp[m][n]);
    }
    return 0;
}