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

这题和建邮局那个题差不多，把输入的数据预处理一下就可以了。

#include<bits/stdc++.h>
using namespace std;
int n,m,a1[1001],f[500][500],mi[500][500],i,j,a[1001];
int main()
{
    cin>>n>>m;
        for(i=1;i<n;i++)
       cin>>a1[i];
       a[1]=0;
       for(i=2;i<=n;i++)
           a[i]=a[i-1]+a1[i-1];
      for(i=1;i<=n;i++)
          for(j=1;j<=m;j++)
                    f[i][j]=9999999;//赋为最大值
              for(i=1;i<=n;i++)
              {
                   for(j=i+1;j<=n;j++)
                    mi[i][j]=mi[i][j-1]+a[j]-a[(i+j)/2];//从第i个村庄到第j个村庄中只建一个小学的状态转移方程
                    f[i][1]=mi[1][i];//状态转移方程的边界

              }

                 for(i=2;i<=n;i++)
                    for(j=2;j<=m;j++)
                        for(int k=j-1;k<i;k++)
                    f[i][j]=min(f[i][j],f[k][j-1]+mi[k+1][i]);
         cout<<f[n][m]<<endl;

         return 0;
}