POJ NOI题库 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。

输出

各村庄到最近学校的距离之和的最小值。

思路：      由数据范围可知，复杂度最多为O(n*n)，最多两重循环，于是用dp[i][j]表示在1...i号小学中放j个学校的距离之和的最小值(i>j)。当i<=j时，dp[i][j] = 0。
    于是有状态转移方程：
    dp[i][j] = min(dp[k][j-1] + s[k][i]) 其中k在区间[1,i)内
s[i][j]指在i号村和j号村中的一个学校到之间各个村的距离和的最小值，即为mid = (i+j)/2号村庄到各个村的距离之和。

#include<cstdio>
#include<cmath>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<sstream>
#include<map>
#include<stack>
using namespace std;
int n,m,dist[501][501]={0},x[500],dp[501][501],s[501][501];
int len(int left,int right){
    int x = 0;
    int mid = (left+right)/2;
    for(int i = left;i<=right;i++) x+=dist[i][mid];
    return x;
}
int main(){
    cin>>m>>n;
    memset(dist, 0, sizeof(dist));
    memset(dp, 8, sizeof(dp));
    x[1] = 0;
    for(int i = 1;i<=m-1;i++){
        int t;
        scanf("%d",&t);
        x[i+1] = x[i] + t;//建立坐标轴，第一个节点为原点
    }
    for(int i = 1;i<=m-1;i++)
        for(int j = i+1;j<=m;j++)
            dist[j][i] = dist[i][j] = fabs(x[j] - x[i]); //结点i、j间的距离
    for(int i = 1;i<=m;i++){
        dp[i][1] = s[1][i];
        dp[i][i] = 0;
        for(int j = 1;j<=m;j++) s[i][j] = len(i,j);
    }
    for(int i = 2;i<=m;i++)
        for(int j = 2;j<=n;j++)
            for(int k = 1;k<i;k++)
                dp[i][j] = min(dp[i][j],dp[k][j-1]+s[k+1][i]);
    printf("%d",dp[m][n]);
    return 0;
}