codves4203山区建小学 dp

题目描述 Description
时间限制: 1 s 空间限制: 16000 KB
政府在某山区修建了一条道路，恰好穿越总共m个村庄的每个村庄一次，没有回路或交叉，任意两个村庄只能通过这条路来往。已知任意两个相邻的村庄之间的距离 为di（为正整数），其中，0 < i < m。为了提高山区的文化素质，政府又决定从m个村中选择n个村建小学（设 0 < n < = m < 500 ）。请根据给定的m、n以及所有相邻村庄的距离，选择在哪些村庄建小学，才使得所有村到最近小学的距离总和最小，计算最小值。

输入描述 Input Description
第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 Description
各村庄到最近学校的距离之和的最小值。
样例输入 Sample Input
10 2
3 1 3 1 1 1 1 1 3
样例输出 Sample Output
18
数据范围及提示 Data Size & Hint
0 < n < = m < 500
题解：设一个f[i][j]，记录在第i个点已经建立j所小学f[k][j-1] k

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int qq=600;    //数组开的够用就行别太大  初始化数组会浪费时间 
int num[qq],sum[qq];
int f[qq][qq],cut[qq][qq];
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));  //以后初始化数组尽量用memset比暴力for一边快 
    for(int i=1;i<=m;i++)
        for(int j=i;j<=m;j++)
        {
            int tot=0;
            int s=(i+j)/2;
            for(int k=i;k<s;k++)
                tot+=sum[s]-sum[k];
            for(int k=s;k<=j;k++)
                tot+=sum[k]-sum[s];
            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=2;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",f[m][n]);
    return 0;
}