summaryrefslogtreecommitdiff
path: root/datastructures/fenwickTree2.cpp
diff options
context:
space:
mode:
authorGloria Mundi <gloria@gloria-mundi.eu>2024-05-02 22:18:05 +0200
committerGloria Mundi <gloria@gloria-mundi.eu>2024-05-02 22:18:05 +0200
commit2ff696b7a72f484f435ffc37dcacedb6fa434b3b (patch)
treecae8bed3e7fecb61f6d0aec582bea5466ea5939b /datastructures/fenwickTree2.cpp
parent2ecbcca9952475c40bfd2a936b7631b236dfc922 (diff)
remove sz() from Fenwick trees
Diffstat (limited to 'datastructures/fenwickTree2.cpp')
-rw-r--r--datastructures/fenwickTree2.cpp14
1 files changed, 7 insertions, 7 deletions
diff --git a/datastructures/fenwickTree2.cpp b/datastructures/fenwickTree2.cpp
index 43c3a32..086e785 100644
--- a/datastructures/fenwickTree2.cpp
+++ b/datastructures/fenwickTree2.cpp
@@ -1,21 +1,21 @@
vector<ll> add, mul;
void update(int l, int r, ll val) {
- for (int tl = l + 1; tl < sz(add); tl += tl&(-tl))
+ for (int tl = l + 1; tl < ssize(add); tl += tl & -tl)
add[tl] += val, mul[tl] -= val * l;
- for (int tr = r + 1; tr < sz(add); tr += tr&(-tr))
+ for (int tr = r + 1; tr < ssize(add); tr += tr & -tr)
add[tr] -= val, mul[tr] += val * r;
}
void init(vector<ll>& v) {
- mul.assign(sz(v) + 1,0);
- add.assign(sz(v) + 1,0);
- for(int i = 0; i < sz(v); i++) update(i, i + 1, v[i]);
+ mul.assign(size(v) + 1, 0);
+ add.assign(size(v) + 1, 0);
+ for(int i = 0; i < ssize(v); i++) update(i, i + 1, v[i]);
}
-ll prefix_sum (int i) {
+ll prefix_sum(int i) {
ll res = 0;
- for (int ti = i; ti > 0; ti -= ti&(-ti))
+ for (int ti = i; ti > 0; ti -= ti & -ti)
res += add[ti] * i + mul[ti];
return res;
}