esraa_syam

fenwick tree

Jul 23rd, 2023 (edited)
199
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.69 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. #define nl "\n"
  3. #define fi first
  4. #define se second
  5. #define pi 3.14159
  6. #define ll long long
  7. #define odd(a)  (a&1)
  8. #define even(a)  !(a&1)
  9. #define Mod 1'000'000'007
  10. #define INF 2'000'000'000
  11. #define sz(x) int(x.size())
  12. #define charToInt(s) (s - '0')
  13. #define ull unsigned long long
  14. #define number_line iota(all(vec) , 1)
  15. #define all(s) s.begin(), s.end()
  16. #define rall(v) v.rbegin() , v.rend()
  17. #define fixed(n) fixed << setprecision(n)
  18. #define Num_of_Digits(n) ((int)log10(n) + 1)
  19. #define to_decimal(bin) stoll(bin, nullptr, 2)
  20. #define Ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  21. #define Floor(n, m) (((n) / (m)) - ((n) % (m) ? 0 : 1))
  22. #define Upper(s) transform(all(s), s.begin(), ::toupper);
  23. #define Lower(s) transform(all(s), s.begin(), ::tolower);
  24. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  25. //  ----- bits-------
  26. #define pcnt(n) __builtin_popcount(n)  
  27. #define pcntll(n) __builtin_popcountll(n)
  28. #define clz(n) __builtin_clz(n)    // <---100
  29. #define clzll(n) __builtin_clzll(n)
  30. #define ctz(n) __builtin_ctz(n)  // 0001---->
  31. #define ctzll(n) __builtin_ctzll(n)
  32.  
  33. using namespace std;
  34.  
  35. template < typename T = int > istream& operator >> (istream &in, vector < T > & v){
  36.     for(auto & x : v) in >> x;
  37.     return in;
  38. }
  39.  
  40. template < typename T = int > ostream& operator << (ostream &out, const vector < T > & v){
  41.     for(const T & x : v) out << x << " ";
  42.     return out;
  43. }
  44. void esraa()
  45. {
  46.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  47. #ifndef ONLINE_JUDGE
  48.     freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  49. #endif
  50. struct fenwick_tree{
  51.     int n{};
  52.     vector < int > bit;
  53.  
  54.     void init(int n){
  55.         this->n = n;
  56.         bit.assign(n + 1, 0);
  57.     }
  58.  
  59.     fenwick_tree(int n = 0){
  60.         init(n);
  61.     }
  62.  
  63.     void add(int idx, int val){
  64.         while(idx <= n){
  65.             bit[idx] += val;
  66.             idx += idx & -idx;
  67.         }
  68.     }
  69.  
  70.     int sum(int idx){
  71.         int ret{};
  72.         while(idx > 0){
  73.             ret += bit[idx];
  74.             idx -= idx & -idx;
  75.         }
  76.         return ret;
  77.     }
  78.  
  79.     int sum(int l, int r){
  80.         return sum(r) - sum(l - 1);
  81.     }
  82.  
  83.     void set(int idx, int val){
  84.         add(idx, val - sum(idx, idx));
  85.     }
  86.  
  87.     void update(int idx, int val){
  88.         add(idx, val);
  89.     }
  90.  
  91.     int get(int idx){
  92.         return sum(idx, idx);
  93.     }
  94.  
  95.    void update(int l, int r, int val){
  96.         add(l, val);
  97.         add(r + 1, -val);
  98.     }
  99.  
  100.     int get(int idx){
  101.         return sum(idx);
  102.  
  103. };
  104. void solve(){
  105.  
  106. }
  107. int main()
  108. {
  109.     esraa();
  110.     int t = 1;
  111.     cin >> t;
  112.     while(t--)
  113.       solve();
  114.     return 0;
  115. }
Advertisement
Add Comment
Please, Sign In to add comment