描述
政府在某山区修建了一条道路,恰好穿越总共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。
输出
各村庄到最近学校的距离之和的最小值。
解析:
dp[i][j]表示前i个村庄修j个小学的最小距离。
dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]); k在j-1到i之间
d[i][j]表示在i到j之间修一座小学的最小距离,该点在i到j中间
#include<stdio.h>
#include<math.h>
#define min(a,b) a<b?a:b
int dp[510][510],d[510][510],a[510];
int sum(int i,int j){//建一所学校的最小距离之和
int mid=(i+j)/2,l,ans=0;
for(l=i;l<=j;l++){
ans+=abs(a[l]-a[mid]);
}
return ans;
}
int main(){
int i,j,k,n,m;
scanf("%d%d",&m,&n);
for(i=2;i<=m;i++){
scanf("%d",&a[i]);
a[i]+=a[i-1];
}
//i到j之间建一所学校的最小距离
for(i=1;i<=m;i++){
for(j=i;j<=m;j++){
d[i][j]=sum(i,j);
}
}
//将已知的dp求出
dp[0][0]=0;
for(i=1;i<=m;i++){
dp[i][1]=d[1][i];
}
//动态规划
for(i=n;i<=m;i++){
for(j=2;j<=n&&j<=i;j++){//j=1时dp[i][j]=d[1][i];
dp[i][j]=dp[i-1][j-1];
for(k=i-2;k>=j-1;k--){
dp[i][j]=min(dp[i][j],dp[k][j-1]+d[k+1][i]);
}
}
}
printf("%d",dp[m][n]);
}