题目链接:点击打开链接
题意:铁路上有n个站点,每个站可以往其他站运送粮草,现在要炸掉m段铁路(两个站点之间为一段)使得粮草补给之和最小,剩余每块连通铁路的粮草补给Strategic Value计算方法见题目,不再累赘。
思路:动态规划问题,用dp[i][j]表示前j个站点炸掉i段得到的最小值。
状态转移方程与前篇POJ 1160 Post Office问题类似:dp[i][j] = min(dp[i - 1][k] + cost[k + 1][j]),i < j且i <= k < j,表示前k个站点炸掉i - 1段,第k个站点与第k + 1个站点之间的铁路炸掉,cost[u][v]表示在连通的前提下,第u个点到第v个点之间的相应Strategic Value。时间复杂度为:O(M*N^2),由于此题数据规模为10^3,肯定会超时,这样就不能按照POJ 1160一样没有优化了。
此题用到的为四边形不等式优化dp,先介绍一下。
在动态规划中,经常遇到如下形式的状态转移方程:
dp[i][j] = min/max(dp[i][k - 1],dp[k][j])+ cost[i][j],i < j且i < k ≤ j,时间复杂度为O(N ^ 3)
上述的dp[i][j]表示某种方案下的最优值,cost[i][j]表示在转移时需要额外付出的代价。
且有如下一些定义、定理:我们再定义K[i][j]表示dp[i][j]取得最优值时对应的下标(即当i < k ≤j 时,若k处的dp值为最优值,则K[i][j] = k)。此时有如下定理
定理2:假如dp满足四边形不等式,那么K单调,可以表示为K[i][j-1] <= K[i][j] <= K[i+1][j]或者K[i - 1][j] <= K[i][j] <= K[i][j + 1]等,此处可以根据需要表示,只要符合K单调即可。
再回来看此题,dp[i][j] = min(dp[i - 1][k] + cost[k + 1][j]),i < j且i <= k < j,和上面的式子很相似,那么cost[i][j]满足两个性质吗?因为都是正数,区间越大,相乘项越多,所以区间包含单调性显而易见;四边形不等式性质不易推导,不过经打表可以看出单调性,所以四边形不等式优化适用此题。
// HDU 2829 Lawrence 运行/限制:171ms/1000ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
#define INF 0x3f3f3f3f
#define LL long long
LL value[1005];
LL sum[1005];//前缀和,用于计算cost
LL cost[1005][1005];//cost[i][j]为从i站台到j站台铁路之间的Strategic Value
LL dp[1005][1005];
int s[1005][1005];
int main(){
int n, m;
while (scanf("%d%d", &n, &m) != EOF && n && m) {
memset(sum, 0, sizeof(sum));
memset(cost, 0, sizeof(cost));
for (int i = 1; i <= n; i++) {
scanf("%lld", &value[i]);
sum[i] = sum[i - 1] + value[i];//计算前缀和
}
for (int i = 1; i < n; i++) {
for (int j = i + 1; j <= n; j++) {
cost[i][j] = cost[i][j - 1] + (sum[j - 1] - sum[i - 1]) * value[j];//计算cost
}
}
/*
固定j为n - 1,得到cost[i][j + 1] - cost[i][j]随i单减,满足凸四边形不等式;
for (int i = 1; i <= n; i++) {
printf("%d\n", cost[i][n] - cost[i][n - 1]);
}
又很容易看出cost关于区间包含关系单调,所以四边形不等式优化dp适用。
*/
memset(dp, INF, sizeof(dp));
for (int i = 1; i <= n; i++) {//先预处理一些边界值
dp[0][i] = cost[1][i];
s[0][i] = 1;
s[i][n + 1] = n - 1;
}
//以下的循环体,当i = 1或者j = n时为边界,上面循环体内就是为了处理这些边界
for (int i = 1; i <= m; i++) {//attack的数量
for (int j = n; j > i; j--) {//1~j的站台;j要大于i
for (int k = s[i - 1][j]; k <= s[i][j + 1]; k++) {//单调性
if (dp[i][j] > dp[i - 1][k] + cost[k + 1][j]) {
dp[i][j] = dp[i - 1][k] + cost[k + 1][j];
s[i][j] = k;
}
}
}
}
printf("%lld\n", dp[m][n]);
}
return 0;
}
POJ 1160也可以利用四边形不等式优化,只不过因为数据范围较小,无优化必要。
初学四边形不等式优化dp算法,相关思路参考自:
在此谢过!!!