HihoCoder1621 : 超市规划（四边形DP优化）（）

超市规划

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

小Hi居住的城市的中轴线恰好是一条马路。沿着这条马路一共坐落有N个居民小区，其中第i个小区距离马路一端的距离是Ai。

现在市政厅决定在这条马路沿线修建K个超市(可以修建在任意实数位置)，并且定义“不方便指数”是所有小区与最近的超市距离的平方之和。

当然市政厅希望“不方便指数”越小越好，请求出该指数的最小值。

输入

第一行包含两个整数N和K。

以下N行每行包含一个整数Ai。

对于50%的数据，1 ≤ N, K ≤ 100

对于100%的数据，1 ≤ N, K ≤ 2000 0 ≤ Ai ≤ 100000

输出

一个实数代表指数的最小值，保留3位小数。

样例输入

4 2
1
2
3
4

样例输出

1.000

有O(n^3)的解法,先知道两个结论:

1，i-j的点中找一个点，使得它到其他点的距离和最小，则这个点可以是下标为中值的点（也有可能一条线段,但是这个点再这条线段上，也满足），即x0=x[(i+j)/2]；可以用反证法来证明。

2，i-j的线段上找一个点，使得它到其他点的距离平方的和最小，则这个点是几何中点，即x0=（xi+..xj）/(j-i+1);可以求偏导数来证明。

由2结论，我们得到如下代码：

其中c[i][j]是i到j中有一个超市时的最优解

则有O(n^3)：

for(i=;i<=n;i++)
 for(j=i+;j<=n;j++){
        double mid=(sum[j]-sum[i-])/(j-i+);
        for(k=i;k<=j;j++){
            c[i][j]+=(a[i]-mid)*(a[i]-mid);
        }
 }

化简：c[i][j]=Σa[i]*a[i]+mid*mid-2*mid*a[i]，则有O（n^2）：

for(i=;i<=n;i++)
     for(j=i+;j<=n;j++){
         double mid=(suma[j]-suma[i-])/(j-i+);
         c[i][j]=(summ[j]-summ[i-])+(j-i+)*mid*mid-*mid*(suma[j]-suma[i-]);
     }

dp[i][j]表示前j个点有i个超市的最优解，则有O(nnk):

for(i=;i<=n;i++) dp[][i]=c[][i];
    for(i=;i<=m;i++){
        s[i][n+]=n;
        for(j=n;j>=;j--){
            for(k=;k<=j;k++)
             if(dp[i][j]>dp[i-][k]+c[k+][j]){
                    s[i][j]=k;
                    dp[i][j]=dp[i-][k]+c[k+][j];
             }
        }
    }

四边形不等式优化：则有O(nk);

 for(i=;i<=n;i++) dp[][i]=c[][i];
    for(i=;i<=m;i++){
        s[i][n+]=n;
        for(j=n;j>=;j--){
            for(k=s[i-][j];k<=s[i][j+];k++)
             if(dp[i][j]>dp[i-][k]+c[k+][j]){
                    s[i][j]=k;
                    dp[i][j]=dp[i-][k]+c[k+][j];
             }
        }
    }

所以O(n^2)就okey辣，鸡冻！

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxn=;
const double inf=;
double dp[maxn][maxn],c[maxn][maxn];
double a[maxn],suma[maxn],summ[maxn],s[maxn][maxn];
int main()
{
    int n,i,j,k,m;
    scanf("%d%d",&n,&m);
    for(i=;i<=n;i++) {
        scanf("%lf",&a[i]);
        suma[i]=suma[i-]+a[i];
        summ[i]=summ[i-]+a[i]*a[i];
    }
    sort(a+,a+n+);
    for(i=;i<=n;i++) {
        suma[i]=suma[i-]+a[i];
        summ[i]=summ[i-]+a[i]*a[i];
    }
    for(i=;i<=n;i++)
     for(j=i+;j<=n;j++){
         double mid=(suma[j]-suma[i-])/(j-i+);
         c[i][j]=(summ[j]-summ[i-])+(j-i+)*mid*mid-*mid*(suma[j]-suma[i-]);
     }
    for(j=;j<=m;j++)
     for(i=;i<=n;i++)
      dp[j][i]=inf;
    for(i=;i<=n;i++) dp[][i]=c[][i];
    for(i=;i<=m;i++){
        s[i][n+]=n;
        for(j=n;j>=;j--){
            for(k=s[i-][j];k<=s[i][j+];k++)
             if(dp[i][j]>dp[i-][k]+c[k+][j]){
                    s[i][j]=k;
                    dp[i][j]=dp[i-][k]+c[k+][j];
             }
        }
    }
    printf("%.3lf\n",dp[m][n]);
    return ;
}