POJ1160

来源:互联网 发布:软件实施过程 编辑:程序博客网 时间:2024/05/16 09:29

用数轴描述一条高速公路,有V个村庄,每一个村庄坐落在数轴的某个点上,需要选择P个村庄在其中建立邮局,要求每个村庄到最近邮局的距离和最小。1 <= V <= 3001 <= P <= 30

 

解题思路:明显的区间DP了。

P=1时:中位数问题

 

问题转化为:将总区间划分成P个区间之和。设状态f[i, j]表示在前j个村庄设立i个邮局。转移方程:

  f[i,j]=min(f[i,j],f[i-1,k]+d[k+1,j]),其中d[i,j]表示在i到j只设立一个邮件时的距离和,将其预处理出来即可

 

代码

#include<stdio.h>#include<math.h>#define min(x,y) (x)<(y)? (x):(y)int main(){    int n,p,i,j,k,t;    int a[310],d[310][310],f[310][310];    scanf("%d%d",&n,&p);    for (i=1;i<=n;i++) scanf("%d",&a[i]);    memset(d,0,sizeof(d));    for (i=1;i<=n-1;i++)      for (j=i+1;j<=n;j++)      {        t=(i+j)/2;        for (k=i;k<=j;k++)          d[i][j]+=abs(a[k]-a[t]);      }    for (i=2;i<=n;i++)      for (j=1;j<=n;j++) f[i][j]=100000000;    f[1][1]=0;    for (i=2;i<=n;i++) f[1][i]=d[1][i];    for (i=2;i<=p;i++)      for (j=i;j<=n;j++)        for (k=i;k<=j;k++)          f[i][j]=min(f[i-1][k]+d[k+1][j],f[i][j]);   printf("%d\n",f[p][n]);   return 0;}