又是长期没刷题的咸鱼,,今天做了道树状数组的题。
这道题一开始我没想用二维的,想着用一维的树状数组,把二维下标重新排列成一维的,比如(x*n+y+1)。但后来发现这样不行,会多算:比如4×4的矩阵,下标0≤x≤3,0≤y≤3,对应一维下标从1对应到16
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
若查找范围是(1,1)<->6 到(2,3)<->12 这个矩阵范围,此种方法会把"9"位置上的数也算上。
所以用二维数组。AC代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
#define ll long long
const int MAX=1030;
int n;
ll c[MAX][MAX];
int lowbit(int x)
{
return x&(-x);
}
void update(int x,int y,int val)
{
for(int i=x;i<=n;i+=lowbit(i))
for(int j=y;j<=n;j+=lowbit(j))
c[i][j]+=val;
}
ll quary(int x,int y)
{
ll ans=0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
ans+=c[i][j];
return ans;
}
int main()
{
int ins;
while(scanf("%d",&ins)==1)
{
if(ins==3)
break;
else if(ins==0)
{
scanf("%d",&n);
memset(c,0,sizeof(c));
continue;
}
else if(ins==1)
{
int x,y,t;
scanf("%d%d%d",&x,&y,&t);
x++;y++;
update(x,y,t);
}
else if(ins==2)
{
int l,b,r,t;
scanf("%d%d%d%d",&l,&b,&r,&t);
l++;b++;r++;t++;
printf("%d\n",quary(r,t)-quary(r,b-1)-quary(l-1,t)+quary(l-1,b-1));//注意!!尤其是l-1,b-1
}
}
return 0;
}