两道DP,四年修一次路

第十一届:山区修路

题目描述

SNJ位于HB省西部一片群峰耸立的高大山地,横亘于A江、B水之间,方圆数千平方公里,相传上古的神医在此搭架上山采药而得名。景区山峰均在海拔3000米以上,堪称"华中屋脊"。SNJ是以秀绿的亚高山自然风光,多样的动植物种,人与自然和谐共存为主题的森林生态区。
SNJ处于中国地势第二阶梯的东部边缘,由大巴山脉东延的余脉组成中高山地貌,区内山体高大,高低不平。 交通十分不便。
最近,HB省决定修一条从YC市通往SNJ风景区的高速公路。经过勘测分析,途中需要经过高度分别为H1,H2,……,Hn的N个山区。由于高低不平,除正常的修路开支外,每段还要多出高度差|Hi - Hi-1|*X万元的斜坡费用。Dr. Kong 决定通过填高一些区域的高度来降低总的费用。当然填高也是需要一些费用的。每填高Y单位,需要付出Y2万元费用。
你能否帮Dr. Kong做出一个规划,通过部分填高工程改造,使得总的费用降下来。


第一行: T 表示以下有T组测试数据( 1≤ T ≤8 )
对每组测试数据,   
第一行:N  X(2 ≤ N ≤100,000   1≤ X ≤100)
第二行:N个整数,分别表示N个区域的高度Hi( 1<=Hi<=100 , i=1…. n)


对每组测试数据,输出占一行,一个整数,即经过部分填高工程改造后的最少费用。

dp[i][j]表示 第i个山区 在第j高度下时,所需要的最少费用
状态转移方程:dp[i][j] = min( dp[i-1][k] + 第i个山区填高的费用 + 第i-1和第i个山区斜坡的花费)

为什么可以这样转移?
因为 题目给出的数据 区域最大高度不过100,而山区最大由100000个,借助数据特点。

#include<bits/stdc++.h>
using namespace std;

const int maxn = 100005;
int t;
int a[maxn];
int dp[maxn][210];
int n,x;
const int inf = 0x3f3f3f3f;

int main(){
	cin>>t;
	while(t--){
		cin>>n>>x;
		for(int i=1;i<=n;i++){
			cin>>a[i];
		}
		memset(dp,inf,sizeof(dp));
		for(int i=a[1];i<=100;i++){
			dp[1][i] = (a[1] - i)*(a[1] - i);
		}
		
		for(int i=2;i<=n;i++){
			for(int j=a[i];j<=100;j++){ //第i个山区 在第j米高度时 
				int cost = (j-a[i])*(j-a[i]); 
				for(int k=a[i-1];k<=100;k++){ //第i-1个山区 在第k米时
					dp[i][j] = min(dp[i][j], dp[i-1][k] +  abs(j - k)*x + cost);
				}
			}
		}
		int ans = inf;
		for(int i=1;i<=100;i++){
			ans = min(ans,dp[n][i]);
		}
		cout<<ans<<endl;
	}
	return 0;
} 

第七届:山区修路

题目描述

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

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

现在Dr. Kong希望找到一个恰好含N个元素的不上升或不下降序列B1,B2,….,Bn,作为修过的路路段的高度。要求:
 | A1-B1| + | A2–B2| + ... + | An-Bn|------>最小

O(n^2)

参考博客:https://blog.csdn.net/m0_37579232/article/details/80426543

这道题和第十一届的类似,只不过dp转移需要改变;
因为ai 最大 有1e9,我们不能再用dp[i]j,这里j最大有1e9,这种方法不可行

而题目给出的数据 ai 虽然大,但是 n小啊。

这道题的正确做法:dp[i][j] 表示第i段在第a[j]高度下的最小费用,前i-1段已经修好了,为什么可以这样?因为题目要求最小花费,我们最大也只能修到和 a[j]数组的最大值相等的高度,也就是a[j]
也就是 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;//状态转移方程: 第i段路修成不超过b[j]的最小花费 等于  i-1段路修成不超过b[j]的最小花费 + 第i短路从a[i]修成b[i]的花费 
        }
    }
 
    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));
    }
}

O(n^3)

O(n^3)的做法 更为直观,但是运行了1700ms会超时。
dp[i][j]表示 第i个点 在高度为b[j]下的最小花费
状态转移方程:第i个点 在第b[j]高度 = min(第i-1个点,在第b[k]高度)
所以三层循环直接递推。

#include<bits/stdc++.h>
using namespace std;

const int maxn = 510;
int t;
int n;
int dp[maxn][maxn][2];
int a[maxn];
int b[maxn];
const int inf = 0x3f3f3f3f;

int main(){
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=1;i<=n;i++){
            cin>>a[i];
            b[i] = a[i]; 
        }
        sort(b+1,b+n+1);
        memset(dp,inf,sizeof(dp));
        for(int i=1;i<=n;i++){
            dp[1][i][0] = abs(a[1] - b[i]);
        }
        
        //第一次dp 正序 
        for(int i=2;i<=n;i++){
            for(int j=1;j<=n;j++){ //第i个点 在第b[j]高度 
                for(int k=1;k<=j;k++){ //第i-1个点 在第b[k]高度 
                    dp[i][j][0] = min(dp[i][j][0],dp[i-1][k][0] + abs(a[i] - b[j])); //状态转移递推
                }
            }
        }

        //第二次  这里超时 k从j~n 最坏时间复杂度O(n^3) 
/*        for(int i=2;i<=n;i++){
            for(int j=1;j<=n;j++){
                for(int k=j;k<=n;k++){
                    dp[i][j][1] = min(dp[i][j][1],dp[i-1][k][1] + abs(a[i] - b[j]));
                }
            }
        }
*/
        reverse(a+1,a+n+1);
        for(int i=1;i<=n;i++){
            dp[1][i][1] = abs(a[1] - b[i]);
        }
        //把数组反转重新 再进行一次第一次dp
        for(int i=2;i<=n;i++){
            for(int j=1;j<=n;j++){
                for(int k=1;k<=j;k++){
                    dp[i][j][1] = min(dp[i][j][1],dp[i-1][k][1] + abs(a[i] - b[j]));
                }
            }
        }
        
        int ans = inf;
        for(int i=0;i<=n;i++){
            if(dp[n][i][1] < ans) ans = dp[n][i][1];
            if(dp[n][i][0] < ans) ans = dp[n][i][0]; 
        }
        cout<<ans<<endl;
    }
    
    return 0;
}

本题小结与想法:

从O(n^3) 优化到O(n^2)的一些理解:
O(n^2)利用单调不减性质:tmp求出了 把第i-1段路修成不超过b[j]的最小花费,tmp找出的是(i-1段路修成不超过b[j]的最小花费)的最优解,边推(i从2~n的过程中)边更新这个最优解,以用来求第i段路修成不超过b[j]的最小花费
O(n3):每次j变化时,都从0开始到j结束找一遍i-1段路修成不超过b[j]的最小花费,可以优化成O(n2)边推边更新。

posted @ 2019-05-03 11:48  fishers  阅读(264)  评论(0编辑  收藏  举报