题意:给出n个1数,是1-n的排列,求区间最大值==区间端点和的区间的个数。
思路:RMQ+分治。1-n的排列,我们就可以记录位置,然后用分治来解决;
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + 10;
int n, num[MAXN], pos[MAXN];
int F_Max[MAXN][30];
void Init() {
for(int i = 1; i <= n; i++) {
F_Max[i][0] = num[i];
}
for(int i = 1; (1 << i) <= n; i++) {
for(int j = 1; j + (1 << i) - 1 <= n; j++) {
F_Max[j][i] = max(F_Max[j][i - 1], F_Max[j + (1 << (i - 1))][i - 1]);
}
}
}
int query(int l, int r) {
int k = (int)(log(double(r - l + 1)) / log((double)2));
return max(F_Max[l][k], F_Max[r - (1 << k) + 1][k]);
}
int dfs(int l, int r) {
if(l >= r)
return 0;
int ans = 0;
int ma = query(l, r);//区间最大
int mid = pos[ma];//最大值在的位置
if(mid - l < r - mid) {//挑小的遍历,防超时
for(int i = l; i < mid; i++) {
int d = ma - num[i];//减去左端点
if(pos[d] > mid && pos[d] <= r)//若右边存在
ans++;
}
} else {
for(int i = mid + 1; i <= r; i++) {
int d = ma - num[i];
if(pos[d] < mid && pos[d] >= l)
ans++;
}
}
return ans + dfs(l, mid - 1) + dfs(mid + 1, r);
}
int main() {
scanf("%d", &n);
for(int i = 1; i <= n; i++)
scanf("%d", &num[i]), pos[num[i]] = i;
Init();
cout << dfs(1, n) << endl;
return 0;
}