7624:山区建小学
-
总时间限制:
- 1000ms 内存限制:
- 65536kB
-
描述
-
政府在某山区修建了一条道路,恰好穿越总共m个村庄的每个村庄一次,没有回路或交叉,任意两个村庄只能通过这条路来往。已知任意两个相邻的村庄之间的距离为di(为正整数),其中,0 < i < m。为了提高山区的文化素质,政府又决定从m个村中选择n个村建小学(设 0 < n < = m < 500 )。请根据给定的m、n以及所有相邻村庄的距离,选择在哪些村庄建小学,才使得所有村到最近小学的距离总和最小,计算最小值。
输入
-
第1行为m和n,其间用空格间隔
第2行为(m-1) 个整数,依次表示从一端到另一端的相邻村庄的距离,整数之间以空格间隔。
例如
10 3
2 4 6 5 2 4 3 1 3
表示在10个村庄建3所学校。第1个村庄与第2个村庄距离为2,第2个村庄与第3个村庄距离为4,第3个村庄与第4个村庄距离为6,...,第9个村庄到第10个村庄的距离为3。
输出
- 各村庄到最近学校的距离之和的最小值。 样例输入
-
10 2 3 1 3 1 1 1 1 1 3
样例输出
-
18
来源
-
元培-From Whf
上代码
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAX=1e7;
int f[510][510],dis[510][510],d[510],s[510][510];//dis[i][j]表示从i到j距离
//s[管辖区起点][管辖区终点]=这片辖区内建一个学校,区内村庄到学校的距离和
int i,j,x,n,m;
int dist(int i,int j)
{
int x=0,mid=(i+j)/2;
for(int k=i;k<=j;k++)
x+=dis[k][mid];
return x;
}
int main()
{
scanf("%d%d",&m,&n);
for(i=2;i<=m;i++)
{//从2开始,方便求各村庄间距离
scanf("%d",&x);
d[i]=d[i-1]+x;
}
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)
{
if(i==j)dis[i][j]=0;
else dis[i][j]=dis[j][i]=abs(d[j]-d[i]);
}//初始化两两距离
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)
s[i][j]=dist(i,j);
//计算一个管辖从i到j村庄的学校到这些村庄的距离和
for(i=1;i<=m;i++)
for(j=1;j<=m;j++)
f[i][j]=MAX;
for(i=1;i<=m;i++)f[i][i]=0;
for(i=1;i<=m;i++)f[i][1]=s[1][i];//只建一个学校的情况
//f的初值
for(i=2;i<=m;i++)//村庄
for(j=2;j<=min(i,n);j++)//学校
for(int k=j-1;k<=i-1;k++)//枚举已有的学校管辖的范围
if(i!=j)f[i][j]=min(f[i][j],f[k][j-1]+s[k+1][i]);
printf("%d",f[m][n]);
}
一开始,我表示我不会做,后来,老师讲同乘积最大,邮局是一样的思想,就可以了……