邮局c++

题目描述

一些村庄建在一条笔直的高速公路边上，我们用一条坐标轴来描述这条公路，每个村庄的坐标都是整数，没有两个村庄的坐标相同。

两村庄的距离定义为坐标之差的绝对值。我们需要在某些村庄建立邮局。

使每个村庄使用与它距离最近的邮局，建立邮局的原则是：所 有村庄到各自使用的邮局的距离总和最小。

输入格式

第一行:n m {表示有n个村庄，建立m个邮局}

第二行:a1 a2 a3 .. an {表示n个村庄的坐标}

输出格式

第一行:l {l表示最小距离总和}

样例

样例输入

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

样例输出

 9

数据范围与提示

1<=村庄数<=300

1<=邮局数<=30

1<=村庄坐标<=10000

代码：

#include<bits/stdc++.h>
using namespace std;
int n,m,a[301];
int f[301][31];

int Abs(int x) {return x<0?-x:x;}

int data(int x,int y)
{
    if (x==y) return 0;
    int mid=(x+y)>>1; int sum=0;
    for (int i=x; i<=y; i++)
        sum+=Abs(a[i]-a[mid]);
    return sum;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    memset(f,0x3f,sizeof(f));
    for (int i=1; i<=n; i++)
        f[i][1]=data(1,i);
    for (int i=2; i<=n; i++)
        for (int j=1; j<=m; j++)
        {
            for (int k=j; k<=i-1; k++)
            f[i][j]=min(f[i][j],f[k][j-1]+data(k+1,i));
        }
    printf("%d",f[n][m]);
    return  0;
}