nrk009

INVERSION COUNT USING BIT

Jul 24th, 2019
129
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.39 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define fast ios_base::sync_with_stdio(0);cin.tie(NULL);cout.tie(NULL)
  4. #define ll long long
  5. #define ld long double
  6. #define pb push_back
  7. #define fe first
  8. #define se second
  9. #define st(x) sort(x.begin(),x.end())
  10. #define rst(x) sort(x.rbegin(), x.rend())
  11. #define all(x) x.begin(),x.end()
  12. long double pi = 3.14159265358979323;
  13.  
  14. const double EPS = 1e-12;
  15. const int N = 2*1e5 + 5;
  16. const int mod = 1e9 + 7;
  17.  
  18. //range update using BIT
  19. ll bit[N],n,a[N];
  20. void update(ll idx,ll val)
  21. {
  22.     while(idx <= n)
  23.     {
  24.         bit[idx] += val;
  25.         idx += idx & (-idx);
  26.     }
  27. }
  28.  
  29. ll query(ll idx)
  30. {
  31.     ll sum = 0 ;
  32.     while(idx > 0)
  33.     {
  34.         sum += bit[idx];
  35.         idx -= idx & (-idx);
  36.     }
  37.     return sum;
  38. }
  39. ll convert()
  40. {
  41.     ll temp[n];
  42.     for(int i = 0 ; i < n ;i++ ){
  43.               temp[i] = a[i];
  44.     }
  45.     sort(temp , temp + n);
  46.     for(int i = 0 ; i < n ; i++ )
  47.     {
  48.         a[i] = lower_bound(temp , temp + n ,a[i]) - temp + 1;
  49.     }
  50.     for(int i = 0 ; i <= n ; i++ )
  51.         bit[i] = 0 ;
  52.     ll ans = 0 ;
  53.  
  54.     for(int i = n - 1 ; i >= 0; i-- )
  55.     {
  56.         ans += query(a[i] - 1);
  57.         update(a[i],1);
  58.     }
  59.     return ans;
  60. }
  61. int main()
  62. {
  63.     fast;
  64.     ll t;
  65.     cin >> t ;
  66.  
  67.     while(t--)
  68.     {
  69.         cin >> n ;
  70.         for(int i = 0 ; i < n ;i++ ){
  71.             cin >> a[i];      
  72.         }
  73.        
  74.         cout << convert() << "\n";
  75.     }
  76.  
  77.  
  78.  
  79.  
  80.  
  81.  
  82.  
  83.  
  84.  
  85.  
  86.  
  87.  
  88.  
  89.  
  90.  
  91.  
  92.    
  93.     return 0;
  94.    
  95. }
Add Comment
Please, Sign In to add comment