Codeforces Round #361 (Div. 2) D. Friends and Subsequences RMQ+二分

链接:

http://codeforces.com/contest/689/problem/D

题意:

给出两个数组,问有多少个区间l,r满足

题解:

先用st表预先处理任意一个区间的最值,然后枚举每个左端点,二分找到满足条件的最近和最远的右端点即可

代码:

31 int n;
32 int a[MAXN], b[MAXN];
33 int st_min[MAXN][32], st_max[MAXN][32];
34 
35 void init() {
36     per(i, 0, n) {
37         st_max[i][0] = a[i];
38         st_min[i][0] = b[i];
39         for (int j = 1; (i + (1 << j) - 1) < n; j++) {
40             st_max[i][j] = max(st_max[i][j - 1], st_max[i + (1 << j - 1)][j - 1]);
41             st_min[i][j] = min(st_min[i][j - 1], st_min[i + (1 << j - 1)][j - 1]);
42         }
43     }
44 }
45 
46 int query_max(int l, int r) {
47     int k = log2(r - l + 1);
48     return max(st_max[l][k], st_max[r - (1 << k) + 1][k]);
49 }
50 
51 int query_min(int l, int r) {
52     int k = log2(r - l + 1);
53     return min(st_min[l][k], st_min[r - (1 << k) + 1][k]);
54 }
55 
56 int main() {
57     ios::sync_with_stdio(false), cin.tie(0);
58     cin >> n;
59     rep(i, 0, n) cin >> a[i];
60     rep(i, 0, n) cin >> b[i];
61     init();
62     ll ans = 0;
63     rep(i, 0, n) {
64         if (a[i] > b[i]) continue;
65         int beg = -1, end = -1;
66         int l = i, r = n - 1;
67         while (l <= r) {
68             int mid = (l + r) / 2;
69             if (query_max(i, mid) == query_min(i, mid)) beg = mid;
70             if (query_max(i, mid) >= query_min(i, mid)) r = mid - 1;
71             else l = mid + 1;
72         }
73         if (beg == -1) continue;
74         l = i, r = n - 1;
75         while (l <= r) {
76             int mid = (l + r) / 2;
77             if (query_max(i, mid) == query_min(i, mid)) end = mid;
78             if (query_max(i, mid) > query_min(i, mid)) r = mid - 1;
79             else l = mid + 1;
80         }
81         ans += end - beg + 1;
82     }
83     cout << ans << endl;
84     return 0;
85 }

 

posted @ 2017-08-17 11:58  Flowersea  阅读(189)  评论(1编辑  收藏  举报