邮局问题(DP)

【问题描述】

　　一些村庄建在一条笔直的高速公路边上，我们用一条坐标轴来描述这条公路，每个村庄的坐标都是整数，没有两个村庄的坐标相同。两个村庄的距离定义为坐标之差的绝对值。我们需要在某些村庄建立邮局。使每个村庄使用与它距离最近的邮局，建立邮局的原则是：所有村庄到各自使用的邮局的距离总和最小。

【输入格式】

　　第一行:n m ，表示有n个村庄，建立m个邮局。
　　第二行:a1 a2 a3 .. an 表示n个村庄的坐标。

【输出格式】

　　第一行:表示最小距离总和。

【输入样例】

10 5
1 2 3 6 7 9 11 22 44 50

【输出样例】

9

【数据范围】

1<=村庄数<=300, 1<=邮局数<=30,
1<=村庄坐标<=10000

需要分组的dp，将每一个邮局所对应的村庄都可以看作为一组，这样就可以设状态函数f(i,j)为前j个村庄建立i个邮局距离总和最小，然后写出状态转移方程f(i,j)=min(f(i-1,k-1)+g[k][j]) （i<=k<=j）g[i][j]意为i到j中每个村庄的距离总和最短距离，可以用中位数计算。边界为f(i,i)=0。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn = 302, inf = 500000000;
int n,m,a[maxn],d[maxn][maxn],g[maxn][maxn];
//f(i,j)前j个村庄建立i个邮局距离总和最小
//f(i,j)=min(f(i-1,k-1)+g[k][j]) i<=k<=j
//g[i][j]i到j中每个村庄的距离总和最短


int calc(int i,int j)
{   
    int x=(i+j)/2,y=(i+j)/2+1,ans1=0,ans2=0; //中位数
    for(int t=i;t<=j;t++)
    {
        ans1+=abs(a[t]-a[x]);
        ans2+=abs(a[t]-a[y]);
    }
    return min(ans1,ans2);
}

int main()
{
    //freopen("1.txt", "r", stdin);
    scanf("%d %d", &n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }

    for(int i=1;i<=n;i++)
    for(int j=i;j<=n;j++)
    {
        g[i][j]=calc(i,j); 
    }

    memset(d,10,sizeof(d));
    for(int i=0;i<=n;i++)
    {
        d[i][i]=0;
    } 
    for(int i=1;i<=m;i++)
    for(int j=i;j<=n;j++)
    {
        int t=inf;
        for(int k=i;k<=j;k++)
        {
            t=min(d[i-1][k-1]+g[k][j],t);
        }
        d[i][j]=t;
    }
    printf("%d",d[m][n]);
    return 0;
}