题目链接:http://poj.org/problem?id=1160
题意:有N个村庄,要在这N个村庄中建K个邮局,求如何布置这些邮局能使得所有村庄到离它最近的那个邮局的距离总和最小。
代码:
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#define INF 10000000
int v[305];
int V,P;
int dp[35][305],dis[305][305];
//dp[i][j]表示前i个邮局建在前j个村庄时的最小距离
//dis[i][j],表示有1个邮局在村庄i与j之间时,村庄i至村庄j到邮局的最短距离总和
void Ini()
{
int i,j,k;
int mid;
memset(dis,0,sizeof(dis));
for(i=1;i<=V;i++)
{
for(j = i+1;j<=V;j++)
{
mid= (i+j)/2;//一个邮局建在中间的那个村庄,能使得所有村庄到该邮局的距离总和最小
for(k=i;k<=j;k++)
dis[i][j] += abs(v[k]-v[mid]);
}
}
}
void DP()
{
int i,j,k;
for(i=2;i<=V;i++)
dp[1][i] = dis[1][i];
for(i=2;i<=P;i++)
for(j=1;j<=V;j++)
dp[i][j] = INF;
for(i=2;i<=P;i++)
for(j=i;j<=V;j++)
for(k=i;k<=j;k++)
{
if(dp[i][j] > dp[i-1][k] + dis[k+1][j])
dp[i][j] = dp[i-1][k] + dis[k+1][j];
}
}
int main()
{
while(scanf("%d%d",&V,&P)!=EOF)
{
for(int i=1;i<=V;i++)
scanf("%d",&v[i]);
Ini();
DP();
printf("%d\n",dp[P][V]);
}
return 0;
}