HDU3480

题意:给你n个数,然后让你分成m个集合,每个集合有一个值(最大值减最小值,然后平方),求整个集合的可能最小值。

思路:因为每个集合里的值只和最大和最小值有关,所以很容易想到先排序,然后用DP可求得解,状态转移方程dp[i][j] = min(dp[i][j] , dp[k][j - 1] + (a[i] - a[k + 1]) ^ 2),j表示数组的下标,i表示集合数,dp为最小值,但是因为n为10000,m为5000,这个复杂度肯定会超时,所以可以用斜率或四边形优化来减小复杂度,式子就不证明了,这里直接引用,另外贴上四边形优化的一个链接,讲的挺详细的点击打开链接

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 10010
#define ll long long
using namespace std;


ll a[MAXN];
ll dp[MAXN][MAXN];
int N, M;
ll q[MAXN];
ll head, tail;


void init()
{
    for (int i = 1; i <= N; i++)
        dp[1][i] = (a[i] - a[1])*(a[i] - a[1]);
}


void solve()
{
    for(int i=2;i<= M;i++)
    {
        head=tail=0;
        q[tail++]=i-1;
        for(int j = i;j <= N;j++)
        {
            while(head+1<tail)
            {
                int p1=q[head];
                int p2=q[head + 1];
                int x1=a[p1 + 1];
                int x2=a[p2 + 1];
                int y1=dp[i - 1][p1] + x1*x1;
                int y2=dp[i - 1][p2] + x2*x2;
                if((y2-y1)<= 2*a[j]*(x2-x1))head++;
                else break;
            }
            int k=q[head];
            dp[i][j]=dp[i-1][k]+(a[j]-a[k+1])*(a[j]-a[k+1]);
            while(head+1<tail)
            {
                int p1=q[tail-2];
                int p2=q[tail-1];
                int p3=j;
                int x1=a[p1+1];
                int x2=a[p2+1];
                int x3=a[p3+1];
                int y1=dp[i-1][p1]+x1*x1;
                int y2=dp[i-1][p2]+x2*x2;
                int y3=dp[i-1][j]+x3*x3;
                if ((y3-y2)*(x2-x1)<=(y2-y1)*(x3-x2))tail--;
                else break;
            }
            q[tail++]=j;
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    int k=0;
    while(T--)
    {
        k++;
        scanf("%d%d",&N,&M);
        for (int i = 1; i <= N; i++)
            scanf("%d", &a[i]);
        sort(a+1,a+N+1);
        init();
        solve();
        printf("Case %d: %d\n",k,dp[M][N]);
    }
    return 0;
}


posted @ 2015-08-02 16:27  __夜风  阅读(169)  评论(0编辑  收藏  举报