pku 1160 Post Office

http://acm.pku.edu.cn/JudgeOnline/problem?id=1160

 

題意：有V個村莊，要在V個村莊上建立S也郵局，使得所有的村莊到他們最近的郵局的距離和最小。（郵局建在村莊裡面，和村莊佔據著同一個點）

 

 

程式碼 ：：

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include <iostream>   
#include <climits>
using  namespace std;

#define  MAXN  305
int weight[MAXN][MAXN],dist[MAXN][31],pos[MAXN];
//dist[i][j]表示有j個郵局控制前i個村莊的最小距離和
//weight[i][j]表示從第i個村莊到第j個村莊由一個郵局控制的最小距離和
int main()
{
 int i,j,k,s,v,min;
 while(scanf("%d%d",&v,&s)!=EOF)
 {
  for(i=1;i<=v;i++)
  {
   scanf("%d",&pos[i]);
  }
  //對輸入排序
  sort(pos+1,pos+v+1);
  memset(weight,0,sizeof(weight));
  //計算從第i個村莊到第j個村莊由一個郵局控制的最小距離和
  for(i=1;i<=v;i++)
  {
   for(j=i; j<=v;j++)
   {
    for(k=i;k<=j;k++)
    {
     weight[i][j]+=abs(pos[k]-pos[(i+j)/2]);
    }
   }
  }
  //第一個郵局控制前i個村莊的最小距離和
  for(i=1;i<=v;i++)
   dist[i][1]=weight[1][i];
  for(i=2;i<=s;i++)
  { 
   for(j=i;j<=v;j++)
   {    
    min=INT_MAX; 
    for(k=i-1;k<=j-1;k++)
    {
     //前i-1個郵局控制前k個村莊的最小距離+從第k+1個村莊到第j個村莊由一個郵局控制的最小距離和
     if(dist[k][i-1]+weight[k+1][j]<min)
      min=dist[k][i-1]+weight[k+1][j];
    } 
    dist[j][i]=min;
   }  
  }
  printf("%d/n",dist[v][s]);
 }
 return 0;
}