Description
政府在某山区修建了一条道路,恰好穿越总共m个村庄的每个村庄一次,没有回路或交叉,任意两个村庄只能通过这条路来往。已知任意两个相邻的村庄之间的距离为di(为正整数),其中,0 < i < m。为了提高山区的文化素质,政府又决定从m个村中选择n个村建小学(设 0 < n < = m < 500
)。请根据给定的m、n以及所有相邻村庄的距离,选择在哪些村庄建小学,才使得所有村到最近小学的距离总和最小,计算最小值。Input
第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。
Output
各村庄到最近学校的距离之和的最小值。
Sample Input
10 2
3 1 3 1 1 1 1 1 3
Sample Output
18
dp:cnt[i][j]表示从第i个村庄到第j个村庄最小的花费,dp[i][j]表示前i个村庄建j所小学的最小距离。
代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=505;
int num[maxn],sum[maxn];
int f[maxn][maxn],cut[maxn][maxn];
int main()
{
int m,n;
scanf("%d%d",&m,&n);
for(int i=2;i<=m;i++)
{
scanf("%d",&num[i]);
sum[i]=sum[i-1]+num[i];
}
int inf=1e9+7;
memset(f,inf,sizeof(f));
for(int i=1;i<=m;i++)
for(int j=i;j<=m;j++)
{
int tot=0;
int mid=(i+j)>>1;
for(int k=i;k<mid;k++)
tot+=sum[mid]-sum[k];
for(int k=mid;k<=j;k++)
tot+=sum[k]-sum[mid];
cut[i][j]=tot;
}
for(int i=1;i<=m;i++)
f[i][1]=cut[1][i];
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
for(int k=1;k<i;k++)
f[i][j]=min(f[i][j],f[k][j-1]+cut[k+1][i]);
printf("%d\n",f[m][n]);
return 0;
}