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

思路

（一）
证明：
给定数轴上n个点，在数轴上的所有点中，中位点离所有点的距离最小（即给定区间内，小学应建在中位点处）
。 。 ， 。 。 。
法一：假设”。”为学校，“，”为小学，此时小学不在中位点处
则将小学向右移动距离为k（移动后仍在中位点左侧），对于小学左侧的点，总距离+2k，对于右侧的点，总距离-3k，总体-k，更优；
以此类推，则可证明位于中位点处距离最近；
特别的，对于偶数个点，中位点可以位于最中间的两个点之间的任意位置（本题只能建在中间两个村庄上）
法二：画图看重复部分，在中位点处重复部分最小；

（二） 关于dp 区间型dp
num[i] 存i号点的坐标，前缀和求；
cnt[i][j] 存从i号节点到j号节点建一所小学的总距离；
for循环下，找到i j之间的中位点mid，然后累加每个点的距离；
注意
在mid左边的点i，距离是num[mid]-num[i];
在mid右边的点i，距离是num[i]-num[mid];
dp[i][j] 存 在前i个点，建j所小学的距离；
初始化：1.初始化极大值 2.dp[i][1]=cnt[1][i];
枚举断点k（1~i）
在前i个点建j所学校=在前k个点建j-1所学校+在k+1到i之间建一所学校；
递推式
dp[i][j]=min(dp[i][j],dp[k][j-1]+cnt[k+1][i]);

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=500+10;
int n,m,tot,x;
int num[maxn],dp[maxn][maxn],cnt[maxn][maxn];
int main(){
    scanf("%d%d",&n,&m);
    for(int i=2;i<=n;i++){
        scanf("%d",&x);
        num[i]=num[i-1]+x;
    //  printf("i %d num %d\n",i,num[i]);
    }
    for(int i=1;i<=n;i++)
    for(int j=i;j<=n;j++){
        int mid=(i+j)/2;
        tot=0;
        for(int k=i;k<mid;k++)
        tot+=(num[mid]-num[k]);
        for(int k=j;k>mid;k--)
        tot+=(num[k]-num[mid]);
        cnt[i][j]=tot;
    // printf("i %d j %d cnt %d\n",i,j,cnt[i][j]);
    }
    memset(dp,0X7f,sizeof(dp));
    for(int i=1;i<=n;i++)
    dp[i][1]=cnt[1][i];
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        for(int k=1;k<=i;k++)
        dp[i][j]=min(dp[i][j],dp[k][j-1]+cnt[k+1][i]);
    //  printf("i %d j %d dp %d\n",i,j,dp[i][j]);
    }

    printf("%d",dp[n][m]);
    return 0;   
}