【JSOI 2009】 Count

【题目链接】

点击打开链接

【算法】

二维树状数组

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 300
#define MAXC 100

int N,M,Q,i,j,opt,x,y,xa,ya,xb,yb,c;
int a[MAXN+][MAXN+];

template <typename T> inline void read(T &x) {
        int f = ; x = ;
        char c = getchar();
        for (; !isdigit(c); c = getchar()) { if (c == '-') f = -f; }
        for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
        x *= f;
}

template <typename T> inline void write(T x) {
    if (x < ) { putchar('-'); x = -x; }
    if (x > ) write(x/);
    putchar(x%+'');
}

template <typename T> inline void writeln(T x) {
    write(x);
    puts("");
}

struct BinaryIndexedTree {
        int bit[MAXC+][MAXN+][MAXN+];
        int lowbit(int x) { return x & (-x); }
        inline void modify(int c,int x,int y,int val) {
                int i,j;
                for (i = x; i <= N; i += lowbit(i)) {
                        for (j = y; j <= M; j += lowbit(j)) {
                                bit[c][i][j] += val;
                        }
                }
        }
        inline int query(int c,int x,int y) {
                int i,j,ret=;
                for (i = x; i >= ; i -= lowbit(i)) {
                        for (j = y; j >= ; j -= lowbit(j)) {
                                ret += bit[c][i][j];
                        }
                }
                return ret;
        }
        inline int query(int c,int xa,int xb,int ya,int yb) {
                return query(c,xb,yb) - query(c,xb,ya-) - query(c,xa-,yb) + query(c,xa-,ya-);
        }
} BIT;

int main() {

        read(N); read(M);
        for (i = ; i <= N; i++) {
                for (j = ; j <= M; j++) {
                        read(a[i][j]);
                        BIT.modify(a[i][j],i,j,);
                }
        }
        read(Q);
        while (Q--) {
                read(opt);
                if (opt == ) {
                        read(x); read(y); read(c);
                        BIT.modify(a[x][y],x,y,-);
                        BIT.modify(c,x,y,);
                        a[x][y] = c;
                }    else {
                        read(xa); read(xb); read(ya); read(yb); read(c);
                        writeln(BIT.query(c,xa,xb,ya,yb));
                }
        }

        return ;

}