Advertisement
esraa_syam

segtree

Jul 9th, 2023 (edited)
123
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.66 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 << nl;
  42.     return out;
  43. }
  44.  
  45. void esraa()
  46. {
  47.    // freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  48.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  49. #ifndef ONLINE_JUDGE
  50.     freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  51. #endif
  52. }
  53. struct segtree{
  54.     int size;
  55.     vector < ll > sums;
  56.  
  57.     void init(int n){
  58.         size = 1;
  59.         while(size < n) size *= 2;
  60.         sums.assign(2 * size, 0LL);
  61.     }
  62.     void build(vector < ll > &a, int x, int lx, int rx){
  63.         if(rx - lx == 1){
  64.             if(lx < sz(a)){
  65.                 sums[x] = a[lx];
  66.             }
  67.             return;
  68.         }
  69.         int m = (lx + rx) / 2;
  70.         build(a, 2 * x + 1, lx, m);
  71.         build(a, 2 * x + 2, m, rx);
  72.         sums[x] = sums[2 * x + 1] + sums[2 * x + 2];
  73.     }
  74.  
  75.     void build(vector < ll > &a){
  76.         build(a, 0, 0, size);
  77.     }
  78.    
  79.     void set(int i, int v, int x, int lx, int rx){
  80.         if(rx - lx == 1){
  81.             sums[x] = v;
  82.             return;
  83.         }
  84.         int m = (lx + rx) / 2;
  85.         if(i < m){
  86.             set(i, v, 2 * x + 1, lx, m);
  87.         }else{
  88.             set(i, v, 2 * x + 2, m, rx);
  89.         }
  90.         sums[x] = sums[2 * x + 1] + sums[2 * x + 2];
  91.     }
  92.  
  93.     void set(int i, int v){
  94.         set(i, v, 0, 0, size);
  95.     }
  96.  
  97.     ll sum(int l, int r, int x, int lx, int rx){
  98.         if(lx >= r or l >= rx) return 0;
  99.         if(lx >= l and rx <= r) return sums[x];
  100.         int m = (lx + rx) / 2;
  101.         ll s1 = sum(l, r, 2 * x + 1, lx, m);
  102.         ll s2 = sum(l, r, 2 * x + 2, m, rx);
  103.         return s1 + s2;
  104.     }
  105.  
  106.     ll sum(int l, int r){
  107.         return sum(l, r, 0, 0, size);
  108.     }
  109. };
  110. void solve(){
  111.     int n, m;
  112.     cin >> n >> m;
  113.     segtree st;
  114.     st.init(n);
  115.     vector < ll > a(n);
  116.     cin >> a;
  117.     st.build(a);
  118.     while(m--){
  119.         int op;
  120.         cin >> op;
  121.         if(op == 1){
  122.             int i, v;
  123.             cin >> i >> v;
  124.             st.set(i, v);
  125.         }else{
  126.             int l, r;
  127.             cin >> l >> r;
  128.             cout << st.sum(l, r) << nl;
  129.         }
  130.     }
  131.  
  132. }
  133. int main()
  134. {
  135.     esraa();
  136.     int t = 1;
  137.    // cin >> t;
  138.     //cin.ignore();
  139.     while(t--)
  140.       solve();
  141.     return 0;
  142. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement