nyoj1251 - 山区修路 - dp（详解）

 

山区修路

时间限制：1000 ms  |  内存限制：65535 KB

难度：2

描述

某山区的孩子们上学必须经过一条凹凸不平的土路，每当下雨天，孩子们非常艰难。现在村里走出来的Dr. Kong决定募捐资金重新修建着条路。由于资金有限，为了降低成本，对修好后的路面高度只能做到单调上升或单调下降。

为了便于修路，我们将整个土路分成了N段，每段路面的高度分别A1，A2,….,An。由于将每一段路垫高或挖低一个单位的花费成本相同，修路的总费用与路面的高低成正比。

现在Dr. Kong希望找到一个恰好含N个元素的不上升或不下降序列B1，B2,….,Bn，作为修过的路路段的高度。要求：

 

         | A1-B1| + | A2–B2| + ... + | An-Bn|------>最小

输入

第一行： K 表示有多少组测试数据。
接下来对每组测试数据：
第1行: N 表示整个土路分成了N段
第2~N+1行： A1 A2 ……AN 表示每段路面的高度
2≤k≤10 0≤Ai≤10000000 0≤N≤500 (i=1,…, N)
所有数据都是整数。 数据之间有一个空格。
数据保证| A1-B1|+| A2-B2|+ ... +| An-Bn|的最小值不会超过109

输出

对于每组测试数据，输出占一行：| A1-B1|+| A2-B2|+ ... +| An-Bn|的最小值。

样例输入

271 3 2 4 5 3 958 6 5 6 2

样例输出

31

来源

第七届河南省程序设计大赛

上传者

516108736

思路：

终于弄懂这道题是什么意思了qwq

起初我想的是简单的遍历，分不递减和不递增两种情况。但是这是不合理的！！！

比如：1 100 50 75

按错误的方法：

50需要修到100，此时变成：1 100 100 75

75需要修到100，此时变成：1 100 100 100

（总计：75）

但实际上我们把100变成50就可以啦= =！

言归正传，我们采用dp的方式，

以不递减为例：首先把输入的数组从小到大排序（b1-bn）

dp[i][j]:=把第i段路修成b[j]高度，所需要的最小花费（这时，前i-1段路都已经搞定啦）

由于是不递减，所以第i段路修成b[j]，那么第i-1段路就只能修成b[1]-b[j]，要求第i段路的最优解，则求:

dp[i][j]=min(dp[i-1][1],dp[i-1][2]……dp[i-1][j]）+abs(a[i]-b[j])

第i-1段路修成b[1],b[2]…b[j]的最小值，加上第i段路修成b[j]所需要的消费

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[505],b[505],n,dp[505][505];
const int INF=0x3f3f3f3f;
int cmp(int a,int b){
    return a>b;
}
int solve(){
    int tmp;
    for(int i=1;i<=n;i++){
        dp[1][i]=abs(a[1]-b[i]);
    }
    for(int i=2;i<=n;i++){
        tmp=INF;
        for(int j=1;j<=n;j++){
            tmp=min(tmp,dp[i-1][j]);//tmp求出了把第i-1段路修成不超过b[j]的最小花费
            dp[i][j]=abs(a[i]-b[j])+tmp;
        }
    }

    int ans=INF;
    for(int i=1;i<=n;i++){
        ans=min(ans,dp[n][i]);
    }
    return ans;
}

int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b+1,b+1+n);
        int ans1=solve();
        sort(b+1,b+1+n,cmp);
        int ans2=solve();
        printf("%d\n",min(ans1,ans2));
    }
}