Advertisement
hoanmalai

Fair Fight

Jun 10th, 2022
955
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.37 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4. typedef long long ll;
  5. const int N = 1e5 + 7;
  6. int n, k;
  7. int c[N], d[N], L[N], R[N];
  8. int m[N][24];
  9. int lg[N];
  10.  
  11. void RMQ_process() {
  12.     for (int i = 0; i<n; i++)
  13.         for (int k = 0; k<=20; k++) m[i][k] = 0;
  14.     for (int i = 0; i < n; i++) m[i][0] = d[i];
  15.     for (int k = 1; (1 << k) <= n; k++)
  16.         for (int i = 0; i + (1 << k) - 1 < n; i++)
  17.             m[i][k] = max(m[i][k - 1], m[i + (1 << (k - 1))][k - 1]);
  18. }
  19.  
  20. int RMQ(int u, int v) {
  21.     int k = lg[v - u + 1];
  22.     return max(m[u][k], m[v - (1 << k) + 1][k]);
  23. }
  24.  
  25. int binary(int pos, int sign, int left, int right, int val) {
  26.     int res = -1;
  27.     while (left < right) {
  28.         int mid = (left + right) / 2;
  29.         int bound = pos + mid * sign;
  30.         int rmqVal = RMQ(min(bound, pos), max(bound, pos));
  31.         if (rmqVal <= val) {
  32.             left = mid + 1;
  33.             res = mid;
  34.         } else {
  35.             right = mid;
  36.         }
  37.     }
  38.     return res;
  39. }
  40.  
  41. ll get(int i, int left, int right, int val) {
  42.     int l = binary(i, -1, 0, i - left + 1, val) + 1;
  43.     int r = binary(i, 1, 0, right - i + 1, val) + 1;
  44.     return 1LL * l * r;
  45. }
  46.  
  47. int main() {
  48.     ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
  49.     int T;
  50.     cin >> T;
  51.     int test = 1;
  52.     while (T--) {
  53.         cin >> n >> k;
  54.         for (int i = 0; i<n; i++) {
  55.             cin >> c[i];
  56.         }
  57.         for (int i = 0; i<n; i++) {
  58.             cin >> d[i];
  59.         }
  60.         for(int i = 2; i <= n; i++)
  61.             lg[i] = lg[i / 2] + 1;
  62.         RMQ_process();
  63.         stack <int> st;
  64.         for (int i = 0; i<n; i++) {
  65.             while (!st.empty() && c[st.top()] <= c[i]) {
  66.                 st.pop();
  67.             }
  68.             if (st.empty())
  69.                 L[i] = 0;
  70.             else L[i] = st.top() + 1;
  71.             st.push(i);
  72.         }
  73.         while (!st.empty()) st.pop();
  74.         for (int i = n-1; i>=0; i--) {
  75.             while (!st.empty() && c[st.top()] < c[i]) {
  76.                 st.pop();
  77.             }
  78.             if (st.empty())
  79.                 R[i] = n-1;
  80.             else R[i] = st.top() - 1;
  81.             st.push(i);
  82.         }
  83.         ll res = 0;
  84.         for (int i = 0; i<n; i++) {
  85.             res += get(i, L[i], R[i], c[i] + k) - get(i, L[i], R[i], c[i] - k - 1);
  86.         }
  87.         cout << "Case #" << test++ << ": " << res << endl;
  88.     }
  89. }
  90.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement