/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 17ms 18.77 MiB
#2 Accepted 33ms 18.859 MiB
#3 Accepted 173ms 19.5 MiB
#4 Accepted 230ms 22.793 MiB
#5 Accepted 568ms 39.051 MiB
#6 Accepted 550ms 39.043 MiB
#7 Accepted 560ms 39.02 MiB
#8 Accepted 572ms 39.062 MiB
#9 Accepted 1081ms 39.863 MiB
#10 Accepted 1115ms 60.758 MiB
#11 Accepted 237ms 19.324 MiB

Code

// PIPRA ||  HABIB
#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>

using namespace __gnu_pbds;
using namespace std;

#define int        long long int
#define pb         push_back
#define all(x)     x.begin(),x.end()
#define allr(x)    x.rbegin(),x.rend()
#define ii         pair<int,int>
#define endl       "\n"

template<typename T>
ostream& operator<<(ostream &os, const vector<T> &v) {
    os << '{';
    for (const auto &x : v) os << " " << x;
        return os << '}';
}
using orderedTree = tree<int, null_type, less<int>, 
rb_tree_tag, tree_order_statistics_node_update>;

const int N = 2e5 + 5;

// seg

int seg[4*N];
int a[N];
vector<vector<int>> tr(4*N);

void build(int i, int low, int high){
    if(low == high){
        seg[i] = a[low];

        tr[i] = {a[low]};
        return;
    }

    int l = (2*i), r = (2*i) + 1;
    int m = (low + high) >> 1;

    build(l, low, m);
    build(r, m+1, high);

    tr[i].resize(tr[l].size() + tr[r].size());
    merge(all(tr[l]), all(tr[r]) , tr[i].begin());
}

int query(int i, int low, int high, int l, int r, int value){
    if(low > r or high < l)     return 0;
    if(low >= l and high <= r){
        int cnt = lower_bound(all(tr[i]) , value) - tr[i].begin();
        return cnt;
    }

    int mid = (low + high) >> 1;
    int li = (2 * i);
    int ri = (2 * i) + 1;

    return  query(li, low, mid, l, r, value) + 
            query(ri, mid+1, high, l, r, value);
}

void pipra(){
    int n;  cin >> n;
    for(int i=1 ; i<=n ; i++)   cin >> a[i];

    build(1, 1, n);

    int q;  cin >> q;
    while(q--){
        int l, m, r;    cin >> l >> m >> r;
        int leftChuto = query(1, 1, n, l, m, a[m]);
        int leftBoro = (m - l + 1) - query(1, 1, n, l, m, a[m]+1);

        int rightchuto = query(1, 1, n, m, r, a[m]);
        int rightBoro = (r - m + 1) - query(1, 1, n, m, r, a[m] + 1);

        int ans = 0;
        ans += leftChuto * rightBoro;
        ans += leftBoro * rightchuto;

        cout << ans << endl;
    }
}

int32_t main(){
    // HABIB
    ios_base::sync_with_stdio(false); 
    cin.tie(NULL); cout.tie(NULL);

    int t;    cin>>t;
    while(t--) {
        pipra();
    }
    return 0 ;
}

Information

Submit By
Type
Submission
Problem
P1082 Roy and Query (Hard Version)
Language
C++17 (G++ 13.2.0)
Submit At
2024-10-11 23:28:56
Judged At
2024-10-11 23:28:56
Judged By
Score
100
Total Time
1115ms
Peak Memory
60.758 MiB