openjudge每日推荐——7624山区建小学

描述
政府在某山区修建了一条道路，恰好穿越总共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 <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 600;//codevs开大了会T最后一个点 
int d[MAXN],s[MAXN][MAXN],num[MAXN],ssum[MAXN];

int f[MAXN][MAXN];//定义f[i][j]为有i个村庄建j个小学要求的答案 
int main()
{
    memset(f,0x3f,sizeof(f));
    memset(s,0x3f,sizeof(s));
    int n,m,x,k;
    /*k是用来分段的，
    1到i村庄这段区间内可以分为前k个村庄一个区间 
    和k+1到i村庄这个区间，然后f[i][j]就可以用前者的区间里建j-1所小学，
    与后者的区间里建1个小学转移过来 
    */ 
    scanf("%d%d",&n,&m);
    for(int i = 2; i <= n; i ++)
    {
        scanf("%d",&x);
        d[i] = d[i-1]+x;//d数组记录的是从起点到各个村庄之间的距离
    }
    //处理出一段村庄只有一个小学的最小区间和 
    for(int i = 1; i <= n; i ++)
    for(int j = i; j <= n; j ++)
    {
        int minn = 0; 
        int mid = (i+j)>>1; //显然取中间数建小学是最优的
        for(int k = i ; k < mid; k ++)//从当前区间的起始位置枚举到中间位置 
        minn += d[mid] - d[k];//记录下前半段的距离 
        for(int k = mid ; k <= j; k ++)//再从中间点枚举到区间末尾位置 
        minn += d[k] - d[mid]; 
        s[i][j] = min(minn,s[i][j]);
    } 
    //如果只建一所小学的情况：
    for(int i = 1; i <= n; i ++)
    f[i][1] = s[1][i]; 
    //写出状态转移方程 
    for(int i = 1; i <= n; i ++)//枚举每个村庄，以后用来取区间
    for(int j = 2; j <= m; j ++)//小学从两个开始建，一个的情况处理过了 
    for(int k = 0; k < i; k ++)//k 不会超过 i  
    {
        f[i][j] = min(f[i][j],f[k][j-1]+s[k+1][i]);//s[k+1][i]就是指在k+1到i这段区间内建一所小学，其他村庄到这里来的最小距离和 ； 
    }
    printf("%d\n",f[n][m]);
    return 0;
}