题目链接
翻译
你在位置 \(1\),然后想要到位置 \(n\),每个位置都有一个高度 \(h[i]\), 你可以从位置 \(i\) 跳到位置 \(j\), 当且仅当以下情况之一满足:
- \(i + 1 = j\)
- \(min(h[i],h[j]) > max(h[i+1..j-1])\) 即 \(i\) 到 \(j\) 这一段里的每个位置的高度都低于 \(h[i]\) 和 \(h[j]\)
- \(max(h[i],h[j]) < min(h[i+1..j-1])\) 即 \(i\) 到 \(j\) 这一段里的每个位置的高度都高于 \(h[i]\) 和 \(h[j]\)
问你从位置 \(1\) 跳到位置 \(n\) 最少需要跳多少次。
题解
刚开始在做这道题的时候,老想着对于每一个 \(i\) 会不会对应特别多的 \(j\)。
但其实我们可以这样想,假设我们现在要考察 \(x\) 能跳到哪个位置 \(y\),我们不妨先假设 \(x+1..y-1\) 这一段是全都大于 \(h[x]\) 和 \(h[y]\) 的,即我们翻译中的第三种情况。
那么我们分两类情况讨论:
- 第一类, \(h_x<=h_y\), 这种情况,因为 \(h_{x+1..y}\) 都是大于 \(h_y\) 的,那么这里的 \(h_x\) 实际上就是 \(y\) 左边最靠右的小于 \(h_y\) 的位置,显然这样的 \(x\) 是只有一个的,因为是 左边最靠右的。
- 第二类, \(h_x>h_y\), 这种情况,我们类似地,可以得到 \(h_y\) 是 \(x\) 右边最靠左的小于 \(h_x\) 的位置, 对于 \(x\) 来说,这样的 \(y\) 也是唯一的。
\(x+1..y-1\) 这一段全都小于 \(h[x]\) 和 \(h[y]\) 的情况则类似, 也同样的是得到 \(x\) 和 \(y\) 对应的唯一的一个能跳到的位置。
经过上面的分析的话,不难发现每个位置 \(x\) 能够跳到的位置 \(y\) 其实最多只有 \(4\) 个。
注意到 左边最靠右的小于/大于 某个数的情况,和 右边最靠左的小于/大于 某个数的情况是一个经典的能用单调栈解决的问题(或者你也可以称呼它为只在一端操作的单调队列)。
这道题的话,每个位置需要从左往右维护上升下降栈,然后从右往左也同样要维护上升下降栈。
因此,对于每个位置 \(x\) 分别处理出来它能跳到的位置 \(y\),然后进行 \(dp\) 的转移即可 (\(dp_i\) 为到达位置 \(i\) 所需要的最少步骤数)。
最后输出 \(dp[n]\)。
代码
#include <bits/stdc++.h>
using namespace std;
const int N = 3e5;
int n;
int h[N + 10],incStack[N + 10],topInc,decStack[N + 10],topDec;
vector<int> g[N + 10];
int dp[N + 10];
int main(){
#ifdef LOCAL_DEFINE
freopen("E://9.AlgorithmCompetition//Visitor.txt","r",stdin);
#endif
ios::sync_with_stdio(0),cin.tie(0);
cin >> n;
for (int i = 1;i <= n; i++){
cin >> h[i];
if (i < n){
g[i].push_back(i);
}
}
topInc = 0,topDec = 0;
for (int i = 1;i <= n; i++){
while (topInc > 0 && h[incStack[topInc]] > h[i]) {
topInc--;
}
//h[incStack[topInc]] <= h[i]
if (topInc > 0){
g[incStack[topInc]].push_back(i);
}
incStack[++topInc] = i;
while (topDec > 0 && h[decStack[topDec]] < h[i]){
topDec--;
}
//h[decStack[topDec]] >= h[i]
if (topDec > 0){
g[decStack[topDec]].push_back(i);
}
decStack[++topDec] = i;
}
topInc = 0,topDec = 0;
for (int i = n;i >= 1; i--){
while (topInc > 0 && h[i] < h[incStack[topInc]]){
topInc--;
}
if (topInc > 0){
g[i].push_back(incStack[topInc]);
}
incStack[++topInc] = i;
while (topDec > 0 && h[i] > h[decStack[topDec]]){
topDec--;
}
if (topDec > 0){
g[i].push_back(decStack[topDec]);
}
decStack[++topDec] = i;
}
for (int i = 1;i <= n; i++){
dp[i] = n + 10;
}
dp[1] = 0;
for (int i = 1;i <= n; i++){
int len = g[i].size();
for (int j = 0;j < len; j++){
int y = g[i][j];
dp[y] = min(dp[y],dp[i] + 1);
}
}
cout << dp[n] << endl;
return 0;
}