Advertisement
Vince14

/<> 1517 (top down seg tree)

Sep 21st, 2023
128
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.23 KB | Source Code | 0 0
  1. #include <iostream>
  2. #include <string>
  3. #include <cstring>
  4. #include <algorithm>
  5. #include <cmath>
  6. #include <vector>
  7. #include <set>
  8. #include <map>
  9. #include <stack>
  10. #include <queue>
  11. #include <deque>
  12. #include <unordered_map>
  13. #include <numeric>
  14. #include <iomanip>
  15. using namespace std;
  16. #define pii pair<long long, long long>
  17. #define ll long long
  18. #define FAST ios_base::sync_with_stdio(false); cin.tie(NULL)
  19. const long long dx[4] = {0, 1, 0, -1}, dy[4] = {1, 0, -1, 0};
  20. const long long dl[2] = {1, -1};
  21. const long long MOD = 1000000007;
  22. const long long MAXN = 500005;
  23.  
  24. int N;
  25. int loc[MAXN], arr[MAXN], assigned[MAXN];
  26. int seg[MAXN * 4];
  27. long long ans = 0;
  28.  
  29. void build_seg(int x, int s, int e){
  30.     if(s == e){
  31.         seg[x] = 0;
  32.         return;
  33.     }
  34.  
  35.     int mid = (s + e)/2;
  36.     build_seg(x * 2, s, mid);
  37.     build_seg(x * 2 + 1, mid + 1, e);
  38.     seg[x] = seg[x * 2] + seg[x * 2 + 1];
  39. }
  40.  
  41. int query(int x, int s, int e, int a, int b){
  42.     if(e < a or s > b) return 0;
  43.     if(a <= s and e <= b) return seg[x];
  44.  
  45.     int mid = (s + e) / 2;
  46.     return query(x * 2, s, mid, a, b) + query(x * 2 + 1, mid + 1, e, a, b);
  47. }
  48.  
  49. void update(int x, int s, int e, int idx, int val){
  50.     if(idx < s or idx > e) return;
  51.     if(s == e){
  52.         seg[x] = val;
  53.         return;
  54.     }
  55.  
  56.     int mid = (s + e) / 2;
  57.     update(x * 2, s, mid, idx, val);
  58.     update(x * 2 + 1, mid + 1, e, idx, val);
  59.     seg[x] = seg[x * 2] + seg[x * 2 + 1];
  60.  
  61. }
  62.  
  63. vector<int> tocomp;
  64. vector<vector<int>> num(MAXN);
  65.  
  66. int main() {
  67.     FAST;
  68.     cin >> N;
  69.     for(int i = 1; i <= N; i++){
  70.         cin >> arr[i];
  71.         tocomp.push_back(arr[i]);
  72.     }
  73.     sort(tocomp.begin(), tocomp.end());
  74.     for(int i = 1; i <= N; i++){
  75.         assigned[i] = (int) (std::lower_bound(tocomp.begin(), tocomp.end(), arr[i]) - tocomp.begin()) + 1;
  76.         num[assigned[i]].push_back(i);
  77.     }
  78.     build_seg(1, 1, N);
  79.     for(int k = 1; k <= N; k++){
  80.         for(auto i : num[k]){
  81.             int q = (i - 1) - query(1, 1, N, 1, i);
  82.             // cout << i << " " << loc[i] << " " << q << "\n";
  83.             ans += q;
  84.             update(1, 1, N, i, 1);
  85.         }
  86.     }
  87.     cout << ans << "\n";
  88. }
  89.  
  90. /*
  91. 5
  92. 5 4 3 2 1
  93. 6
  94. 2 1 3
  95. 2 1 4
  96. 1 5 3
  97. 2 3 5
  98. 1 4 3
  99. 2 3 5
  100.  */
  101.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement