牛客网暑期ACM多校训练营（第二场)J farm (二维树状数组）

题目链接：

https://www.nowcoder.com/acm/contest/140/J

思路：

都写在代码注释里了，非常好懂。。

for_each函数可以去看一下，遍历起vector数组比较方便，用for(int i = 0;i < q[i].size();i++)的话，是会有一些弊端的，虽然对于这道题应该没什么影响，但最好规范下。

耗时：2072ms

实现代码：

#include<bits/stdc++.h>
using namespace std;
const int M = 1e6+;
vector<int>t[M];
int n,m;
vector<int>bit[M];
int lowbit(int x){
    return x&-x;
}
 
struct node{
    int a,b,c,d;
    node(){}
    node(int a1,int b1,int c1,int d1):a(a1),b(b1),c(c1),d(d1){}
};
 
struct node1{
    int i,j;
    node1(){}
    node1(int i1,int j1):i(i1),j(j1){}
};
 
vector<node1>k[M];
vector<node>q[M];
 
void add(int x,int y,int z){
    for(int i=x;i<=n;i+=lowbit(i)){
        for(int j=y;j<=m;j+=lowbit(j))
            bit[i][j]+=z;
    }
}
void update(int x1,int y1,int x2,int y2,int z){
    add(x1,y1,z);
    add(x2+,y2+,z);
    add(x1,y2+,-z);
    add(x2+,y1,-z);
}
int sum(int x,int y){
    int res=;
    for(int i=x;i;i-=lowbit(i)){
        for(int j=y;j;j-=lowbit(j)){
            res+=bit[i][j];
        }
    }
    return res;
}
 
template <class T>
inline void scan_d(T &ret)
{
    char c;
    ret = ;
    while ((c = getchar()) < '' || c > '');
    while (c >= '' && c <= '')
    {
        ret = ret *  + (c - ''), c = getchar();
    }
}
 
template <class T>
inline void print_d(T x)
{
    if (x > )
    {
        print_d(x / );
    }
    putchar(x %  + '');
}
 
void fun1(node now){
    update(now.a,now.b,now.c,now.d,);
}
 
void fun2(node now){
    update(now.a,now.b,now.c,now.d,-);
}
 
int num;
void fun3(node1 now){
    if(sum(now.i,now.j)) num++;
}
 
int main()
{
    int t,x,a,b,c,d,z;
    scan_d(n); scan_d(m); scan_d(t);
    for(int i = ;i <= n;i ++) bit[i].resize(m+); //预开空间
    for(int i = ;i <= n;i ++) {
        for(int j = ;j <= m;j ++){
            scan_d(x);
            k[x].push_back(node1(i,j));  //需要x种类药的花的坐标
        }
    }
    for(int i = ;i <= t;i ++){
        scan_d(a);scan_d(b),scan_d(c),scan_d(d);scan_d(z);
        update(a,b,c,d,);  //标记代表这个区间被z种类药撒了
        q[z].push_back(node(a,b,c,d)); //存下z种类药一共撒了哪些区间
    }
    num = ;
    for(int i = ;i <= n*m;i ++){ //遍历所有种类的药
        if(k[i].size()){  //存在需要ki种类药的花
            for_each(q[i].begin(),q[i].end(),fun2); //将ki种类药撒的区间造成的影响全部清0
            for_each(k[i].begin(),k[i].end(),fun3);  //遍历需要k种类药的所有花的坐标如果这个坐标依旧为1，那么代表撒在它上面的并不是k种类的药，这朵花会死亡,num++;
            for_each(q[i].begin(),q[i].end(),fun1);  //再将ki种类药撒的区间还原
        }
    }
    print_d(num);
    printf("\n");
    return ;
}