Update fenwick_tree_2d.cpp
This commit is contained in:
parent
a5b7c15230
commit
b0bc5023ca
1 changed files with 14 additions and 14 deletions
|
@ -1,21 +1,21 @@
|
||||||
class fenwick_tree_2d {
|
class fenwick_tree_2d {
|
||||||
private: int N, M, vector<vector<int>> FT;
|
private: int N, M, vector<vector<int>> FT;
|
||||||
public:
|
public:
|
||||||
fenwick_tree_2d(int n, int m) {
|
fenwick_tree_2d(int n, int m) {
|
||||||
N = n + 1, M = m + 1;
|
N = n + 1, M = m + 1;
|
||||||
FT.resize(N, 0);
|
FT.resize(N, 0);
|
||||||
for (int i = 0; i < N; i++) FT[i].resize(M, 0);
|
for (int i = 0; i < N; i++) FT[i].resize(M, 0);
|
||||||
}
|
}
|
||||||
void update(int x, int y, int val) {
|
void update(int x, int y, int val) {
|
||||||
for (int i = x; i < N; i += i & -i) {
|
for (int i = x; i < N; i += i & -i) {
|
||||||
for (int j = y; j < M; j += j & -j) FT[i][j] += val;
|
for (int j = y; j < M; j += j & -j) FT[i][j] += val;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
int query(int x, int y) {
|
int query(int x, int y) {
|
||||||
int ret = 0;
|
int ret = 0;
|
||||||
for (int i = x; i > 0; i -= i & -i) {
|
for (int i = x; i > 0; i -= i & -i) {
|
||||||
for (int j = y; j > 0; j -= j & -j) ret += FT[i][j];
|
for (int j = y; j > 0; j -= j & -j) ret += FT[i][j];
|
||||||
}
|
}
|
||||||
return ans;
|
return ret;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue