以前写过一个二维线段树,现在忘得差不多了,又想了一遍,就是把一维的每一个节点再抽象成一维的线段树。
这个题查询的时候,每一个x维都要进行y维查询,只要经过的是1的都要加上,因为这样经过的肯定都包含要查询的点
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn=1010;
int N,M,x1,x2,Y1,y2,ans;
struct IntervalTree
{
int cnt[maxn<<2][maxn<<2];
void build()
{
memset(cnt,0,sizeof(cnt));
}
void updateY(int xo,int o,int l,int r)
{
if(Y1<=l&&r<=y2)
{
cnt[xo][o]=!cnt[xo][o];
return;
}
int mid=(l+r)>>1;
if(Y1<=mid)updateY(xo,o<<1,l,mid);
if(y2>mid)updateY(xo,o<<1|1,mid+1,r);
}
void updateX(int o,int l,int r)
{
if(x1<=l&&r<=x2)
{
updateY(o,1,1,N);
return;
}
int mid=(l+r)>>1;
if(x1<=mid)updateX(o<<1,l,mid);
if(x2>mid)updateX(o<<1|1,mid+1,r);
}
void queryY(int xo,int o,int l,int r)
{
if(cnt[xo][o])ans++;
if(l==r)return;
int mid=(l+r)>>1;
if(Y1<=mid)queryY(xo,o<<1,l,mid);
else queryY(xo,o<<1|1,mid+1,r);
}
void queryX(int o,int l,int r)
{
queryY(o,1,1,N);
if(l==r)return;
int mid=(l+r)>>1;
if(x1<=mid)queryX(o<<1,l,mid);
else queryX(o<<1|1,mid+1,r);
}
}tree;
int main()
{
freopen("in.txt","r",stdin);
int T;
char op[5];
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&N,&M);
tree.build();
while(M--)
{
scanf("%s",op);
if(op[0]=='C')
{
scanf("%d%d%d%d",&x1,&Y1,&x2,&y2);
tree.updateX(1,1,N);
}
else
{
scanf("%d%d",&x1,&Y1);
ans=0;
tree.queryX(1,1,N);
printf("%d\n",ans%2);
}
}
if(T)printf("\n");
}
return 0;
}
也可以用二维树状数组写
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define N 1005
int c[N][N],n;
int lowbit(int i){
return i&(-i);
}
void update(int x,int y,int val){
int i,j;
for(i=x;i<=n;i+=lowbit(i))
for(j=y;j<=n;j+=lowbit(j))
c[i][j]+=val;
return ;
}
int getsum(int x,int y){
int s,i,j;
s=0;
for(i=x;i>=1;i-=lowbit(i))
for(j=y;j>=1;j-=lowbit(j))
s+=c[i][j];
return s;
}
int main(){
int tt,m,x1,x2,y1,y2;
char ch[4];
scanf("%d",&tt);
while(tt--){
memset(c,0,sizeof(c));
scanf("%d%d",&n,&m);
getchar();
while(m--){
scanf("%s",ch);
if(ch[0]=='C'){
scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
x2++;
y2++;
update(x2,y2,1);
update(x1,y1,1);
update(x1,y2,-1);
update(x2,y1,-1);
}else
if(ch[0]=='Q'){
scanf("%d%d",&x1,&y1);
printf("%d\n",getsum(x1,y1)&1);
}
}
printf("\n");
}
return 0;
}