2019牛客暑期多校训练营(第一场)A Equivalent Prefixes 单调栈
程序员文章站
2022-04-01 15:41:29
...
题目链接 https://ac.nowcoder.com/acm/contest/881/A
题意:给你2个数组,求最大的p使得 2个数组1到p的任何一个子区间最小值下标相同
题解:设1到p的最小值下标为x,仅需满足x+1到p 2个数组对应值的大小排序相同
使用单调栈从小到大排:最开始只有一列,是最小的,如果后面有比栈底元素小的,栈会被弹空,把最小的压入栈,如果后面来的数据在栈中排序不同的话,2个栈弹栈次数不同,比较2个栈数据的数量就可以判断当前加入的值满不满足
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <map>
#include <string>
#include <cstring>
#include <queue>
#include <stack>
#include <cmath>
using namespace std;
const int N = 1e5 + 5;
stack<int> Ma;
stack<int> Mb;
int a[N];
int b[N];
int n;
void initMa() {
while (!Ma.empty()) Ma.pop();
}
void initMb() {
while (!Mb.empty()) Mb.pop();
}
int main() {
while (cin>>n) {
for (int i = 0; i < n; i++) scanf("%d", a + i);
for (int i = 0; i < n; i++) scanf("%d", b + i);
if (!Ma.empty()) initMa();
if (!Mb.empty()) initMb();
Ma.push(a[0]);
Mb.push(b[0]);
int k = 1;
for (int i = 1; i < n; i++) {
while (1) {
if (Ma.empty() || Ma.top() < a[i]) {
Ma.push(a[i]);
break;
}
else Ma.pop();
}
while (1) {
if (Mb.empty() || Mb.top() < b[i]) {
Mb.push(b[i]);
break;
}
else Mb.pop();
}
if (Ma.size() != Mb.size()) break;
k++;
}
cout << k << endl;
}
return 0;
}