Ural 1031

题意：在一条铁路上有N个火车站，只有三种票，求两个车站之间的最少花费

方法：一开始以为是一道简单题，很NC的开了一个二位数组，很可惜，MLE了，而且中间还很SB的想到了用记忆化搜索可能比构建更快一些。。。

           经过大牛们的悉心指导，原来用一维就可以搞定 转移方程为f[j] = min(f[i] + GetCost(i,j)) （start <= i < j ,  start + 1 < j <=end) 不优化可能要超时，所以可以从j-1到start的方向寻找i ，中间不行即可break，算是一个剪枝吧。

代码： O（n^2)

 

#include <iostream> #include <algorithm> #include <cstring> #include <limits> using namespace std; #define MAX 11000 const int INT_MAX1 = numeric_limits<int>::max(); int L1, L2, L3; int C1, C2, C3; int istart, iend; int N; int dis[MAX]; int f[MAX]; void Input() { cin >> L1 >> L2 >> L3; cin >> C1 >> C2 >> C3; cin >> N; cin >> istart >> iend; dis[1] = 0; for (int i = 2; i <= N; i++)//要 <= 才是N-1个 cin >> dis[i]; } int GetCost(int i, int j) { int tdis = dis[j] - dis[i]; if (tdis <= L1) return C1; if (tdis <= L2) return C2; if (tdis <= L3) return C3; return -1; } void DP() { int cost; for (int j = istart + 1; j <= iend; j++) { f[j] = INT_MAX1; for (int i = j - 1; i >= istart; i--) { if ((cost = GetCost(i, j)) != -1) f[j] = min(f[i] + cost, f[j]); else break; } } } int main() { Input(); if (istart > iend){//最为阴险的地方 swap(istart, iend); DP(); cout << f[iend] << endl; } else if (istart == iend) cout << 0 << endl; else { DP(); cout << f[iend] << endl; } return 0; }

 

其实还有O（n）的算法，关键在于寻找f[i]的时候只需要找三个地方

 

用一个p[i][j] 表示距离车站i 用第j中票能买到的最近的车站

这样转移方程就变成了f[j] = min(f[ p[j][k] ] + cost[j])

 

注意两个地方：第一个是计算P的时候需要istart和iend，所以要先交换再计算。第二个地方比较淫荡，当p[j][k]==j的时候，

f[j] > f[ p[j][k] ] + cost[k] 可能成立，因为f[j]开始赋值为INT_MAX，溢出了。。

 

代码：

 

 

#include <iostream> #include <algorithm> #include <cstring> #include <limits> using namespace std; #define MAX 11000 const int INT_MAX1 = numeric_limits<int>::max(); int len[3]; int cost[3]; int istart, iend; int N; int dis[MAX]; int f[MAX]; int p[MAX][3]; void Input() { for (int i = 0; i < 3; i++) cin >> len[i]; for (int i = 0; i < 3; i++) cin >> cost[i]; cin >> N; cin >> istart >> iend; dis[1] = 0; for (int i = 2; i <= N; i++)//要 <= 才是N-1个 cin >> dis[i]; } void ComputeP() { for (int k = 0; k < 3; k++)//票种类 { int pos = iend - 1; for (int j = iend; j >= istart; j--) { while (dis[j] - dis[pos] <= len[k] && pos >= istart) pos--; p[j][k] = pos + 1; } } } void DP() { for (int j = istart + 1; j <= iend; j++) { f[j] = INT_MAX1; for (int k = 0; k < 3; k++) { if (j != p[j][k] && f[j] > f[ p[j][k] ] + cost[k]){ f[j] = f[ p[j][k] ] + cost[k]; } } } } int main() { Input(); if (istart > iend) {//最为阴险的地方 swap(istart, iend); ComputeP(); DP(); cout << f[iend] << endl; } else if (istart == iend) cout << 0 << endl; else { ComputeP(); DP(); cout << f[iend] << endl; } return 0; }