Advertisement
Josif_tepe

Untitled

May 9th, 2024
718
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.11 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. const int maxn = 2e5 + 10;
  4. int next_idx[maxn];
  5.  
  6. vector<int> segment_tree[3 * maxn];
  7. vector<int> merge_two_nodes(vector<int> & a, vector<int> & b) {
  8.     int i = 0, j = 0;
  9.     vector<int> c;
  10.     while(i < (int) a.size() and j < (int) b.size()) {
  11.         if(a[i] <= b[j]) {
  12.             c.push_back(a[i]);
  13.             i++;
  14.         }
  15.         else {
  16.             c.push_back(b[j]);
  17.             j++;
  18.         }
  19.     }
  20.     while(i < (int) a.size()) {
  21.         c.push_back(a[i]);
  22.         i++;
  23.     }
  24.     while(j < (int) b.size()) {
  25.         c.push_back(b[j]);
  26.         j++;
  27.     }
  28.     return c;
  29. }
  30. void build_tree(int L, int R, int node) {
  31.     if(L == R) {
  32.         segment_tree[node].push_back(next_idx[L]);
  33.     }
  34.     else {
  35.         int middle = (L + R) / 2;
  36.         build_tree(L, middle, 2 * node);
  37.         build_tree(middle + 1, R, 2 * node + 1);
  38.         segment_tree[node] = merge_two_nodes(segment_tree[2 * node], segment_tree[2 * node + 1]);
  39.     }
  40. }
  41. // L R i L R j L R
  42. int query(int L, int R, int node, int i, int j) {
  43.     if(R < i or j < L) {
  44.         return 0;
  45.     }
  46.     if(i <= L and R <= j) {
  47.         return (int) segment_tree[node].size() - (lower_bound(segment_tree[node].begin(), segment_tree[node].end(), j + 1) - segment_tree[node].begin());
  48.     }
  49.     int middle = (L + R) / 2;
  50.     return query(L, middle, 2 * node, i, j) + query(middle + 1, R, 2 * node + 1, i, j);
  51. }
  52. int main() {
  53.     ios_base::sync_with_stdio(false);
  54.     int n, q;
  55.     cin >> n >> q;
  56.     for(int i = 0; i < n; i++) {
  57.         next_idx[i] = n;
  58.     }
  59.     vector<pair<int, int>> v(n);
  60.     for(int i = 0; i < n; i++) {
  61.         cin >> v[i].first;
  62.         v[i].second = i;
  63.     }
  64.     sort(v.begin(), v.end());
  65.  
  66.     for(int i = 0; i < n - 1; i++) {
  67.         if(v[i].first == v[i + 1].first) {
  68.             next_idx[v[i].second] = v[i + 1].second;
  69.         }
  70.     }
  71.     build_tree(0, n - 1, 1);
  72.    
  73.    
  74.     for(int i = 0; i < q; i++) {
  75.         int a, b;
  76.         cin >> a >> b;
  77.         a--; b--;
  78.  
  79.         cout << query(0, n - 1, 1, a, b) << endl;
  80.     }
  81.  
  82.    
  83.     return 0;
  84. }
  85.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement