A. Equivalent Prefixes
题目链接:https://ac.nowcoder.com/acm/contest/881/A
题意:给定两个数组 A 和 B,每个数组中的值不重复。求一个最大下标 P,使得 A 数组区间 [1, P] 与 B 数组 [1, P] 的相同子区间的区间最小值的下标相同。
思路:假设题目求的区间 叫 合法区间
- 假如 A 中一个值 a[i] 左边第一个比 a[i] 小的下标为 0、右边第一个比 a[i] 小的下标为 RightA,那么a[i] 是最小值的范围是 [1,RightA)
- 同理:B 中一个值 b[i] 左边第一个比 b[i] 小的下标为 0、右边第一个比 b[i] 小的下标为 RightB,那么b[i] 是最小值的范围是 [1,RightB)
- 显然 区间 min(RightA,RightB) 往后的区间一定不合法。
- 假设一开始 P = n,每一次都可以根据以上确定一个最大合法的右端点,所以取这个右端点的最小值即是答案.
AC代码:
#include<bits/stdc++.h>
#define debug(x) cout << "[" << #x <<": " << (x) <<"]"<< endl
#define pii pair<int,int>
#define clr(a,b) memset((a),b,sizeof(a))
#define rep(i,a,b) for(int i = a;i < b;i ++)
#define pb push_back
#define MP make_pair
#define LL long long
#define ull unsigned LL
#define ls i << 1
#define rs (i << 1) + 1
#define INT(t) int t; scanf("%d",&t)
using namespace std;
const int maxn = 1e5 + 10;
int a[maxn],b[maxn];
int RightA[maxn];//,LeftA[maxn];
int RightB[maxn];//,LeftB[maxn];
stack<int> St;
int main() {
int n;
while(~scanf("%d",&n)){
for(int i = 1;i <= n;++ i) scanf("%d",&a[i]);
for(int i = 1;i <= n;++ i) scanf("%d",&b[i]);
while(!St.empty())
St.pop();
for(int i = 1; i <= n; ++ i) {
while(!St.empty() && a[St.top()] > a[i]) {
RightA[St.top()] = i;
St.pop();
}
St.push(i);
}
while(!St.empty()) {
RightA[St.top()] = n + 1;
St.pop();
}
for(int i = 1; i <= n; ++ i) {
while(!St.empty() && b[St.top()] > b[i]) {
RightB[St.top()] = i;
St.pop();
}
St.push(i);
}
while(!St.empty()) {
RightB[St.top()] = n + 1;
St.pop();
}
int ans = n;
for(int i = 1;i <= n;++ i){
int tmpa = RightA[i] - 1;
int tmpb = RightB[i] - 1;
if(tmpa != tmpb)
ans = min(ans,min(tmpa,tmpb));
}
printf("%d\n",ans);
}
return 0;
}