题意
给定n个村庄的坐标,选m个地方建邮局,要求各个村庄到其最近的邮局的距离之和最小。距离定义为坐标之差。
解题
设dp[i][j]表示前i个村庄建j个邮局的最小距离和。
先预处理对于每个区间建一个邮局的最小距离和。设dis[i][j]表示在区间[i][j]建一个邮局的最小距离和。dis[i][j]=dis[i][j-1]+a[j]-a[(i+j)/2].因为建一个邮局的话,显然建立在中位数上是最优的,画个图就明了了。多一个村庄,虽然邮局的位置可能变了,但是前i-1个距离之和是不变的。所以多出了的距离就是第j个村庄到邮局的距离。
考虑加一个邮局:
前k个村庄建j-1个邮局,在第k+1个村庄到第i个村庄之间建一个邮局。
dp[i][j]=max{dp[k][j-1]+dis[k+1][j]},k是决策。取值范围为[j-1,i-1].
AC代码
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <iostream>
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=330;
int dp[maxn][maxn];
int a[maxn];
int dis[maxn][maxn];//dis[i][j]表示在第i个村和第j个村之间建一个邮局的“代价”
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
//预处理dis[i][j]
for(int i=1;i<=n;i++)
for(int j=i+1;j<=n;j++)
{
//显然在[i,j]放一个邮局,放在最中间是最优的
dis[i][j]=dis[i][j-1]+a[j]-a[(i+j)/2];
}
for(int i=1;i<=n;i++)
{
dp[i][i]=0;
dp[i][1]=dis[1][i];
}
for(int j=2;j<=m;j++)
{
for(int i=j+1;i<=n;i++)
{
dp[i][j]=INF;
for(int k=j-1;k<i;k++)
{
dp[i][j]=min(dp[i][j],dp[k][j-1]+dis[k+1][i]);
}
}
}
printf("%d\n",dp[n][m]);
}
return 0;
}