牛客国庆day2-J-farm(二维树状数组)

题意

有一个n*m的矩阵，每个格子都有它需要撒的药的编号，现在有t次从(x1,y1)-(x2,y2)的撒药k，然后如果撒的药的编号与它所需要的不一样的话，就该格子的花死亡。

思路

这题主要是想总结一下（二维树状数组）。
vector-存树状数组-存该药撒的范围-存这种药需要撒的格子坐标。
我们通过树状数组来将撒的范围的±1，来标记这次撒药，通过遍历来找这个格子的标记是否为0，因为如果他就被他自己所需要的药撒过的话，就1->0，但如果别人撒过它或者没药来撒它，就-/>0
剩下的写注释了。
二维树状数组

运用了差分的思想，来进行update，循环原理不变，从一维到二维而已。

#include<bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
typedef long long ll;
typedef pair<ll,ll> PII;
const int N=1e6+5;
unsigned int read()//快读
{
    unsigned int ret=0;
    char ch=getchar();
    while(ch>'9'||ch<'0')ch=getchar();
    while(ch>='0'&&ch<='9')
    {
        ret=ret*10+ch-'0';
        ch=getchar();
    }
    return ret;
}
struct node
{
    ll x1,y1,x2,y2;
};

vector<ll>bt[N];
ll n,m,t;
ll lowbit(ll x){
    return x&-x;
}
void add(ll x,ll y,ll k)
{
    for(int i=x; i<=n; i+=lowbit(i))
    {
        for(int j=y; j<=m; j+=lowbit(j))
        {
            bt[i][j]+=k;//类似于一维的update更新
        }
    }
}
void update(ll x1,ll y1,ll x2,ll y2,ll k)
{
    add(x1,y1,k);
    add(x2+1,y2+1,k);//差分
    add(x1,y2+1,-k);
    add(x2+1,y1,-k);
}
ll query(ll x,ll y)
{
    ll res=0;
    for(int i=x; i; i-=lowbit(i))
    {
        for(int j=y; j; j-=lowbit(j))
        {
            res+=bt[i][j];//类似于一维的query
        }
    }
    return res;
}
int main ()
{
    //IOS;
    //cin>>n>>m>>t;
    scanf("%lld %lld %lld",&n,&m,&t);
    for(int i=0;i<=n;i++) bt[i].resize(m+2);//第一次知道vector需要提前开空间
    vector<vector<node>> v(n*m+2);
    vector<vector<PII>> mp(n*m+2);//数组超限，减内存的
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=m; j++)
        {
            ll x;
            scanf("%lld",&x);
            mp[x].push_back(make_pair(i,j));
        }
    }
    while(t--)
    {
        ll x1,y1,x2,y2,k;
        x1=read();y1=read();x2=read();y2=read();k=read();
       // cin>>x1>>y1>>x2>>y2>>k;
        //cout<<x1<<" "<<y1<<" "<<x2<<" "<<y2<<k<<endl;
        v[k].push_back({x1,y1,x2,y2});
        update(x1,y1,x2,y2,1);//+1
    }
    ll ans=0;
    for(int i=1; i<=n*m; i++)
    {
        if(mp[i].size())
        {
            for(auto j:v[i])
            {
                update(j.x1,j.y1,j.x2,j.y2,-1);//消除-1
            }
            for(auto j:mp[i])
            {
                ll p=query(j.first,j.second);
                if(p!=0)
                {
                    ans++;
                }
            }
            for(auto j:v[i])
            {
                update(j.x1,j.y1,j.x2,j.y2,1);//记得加回来
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}