HDU - 4456 Crowd

题目见：
[http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=34034]

一道显然的二维树状数组，还是单点修改，矩阵查询。
重点是把曼哈顿距离包含的一个立着的正方形旋转45度，变成水平的。
虽然同时mat由水平变成了立着的，但计算时并不会受影响，因为超出的地方不会被修改。
有两个地方要注意：
(1).坐标的转换
(2).下标离散化

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define MAXM  80000
#define MAXN 5000000
int n,m,p[MAXN+10],x[MAXN+10],y[MAXN+10],z[MAXN+10];
int h[MAXN+10],cnt,c[MAXN+10],w;
int lowbit(int x)
{
    return x&(-x);
}
void Get_ready(int xx,int yy)
{
    for(int i=xx; i<=w; i+=lowbit(i))
        for(int j=yy; j<=w; j+=lowbit(j))
            h[cnt++]=i*w+j;
}
void GetPoint(int xx,int yy,int &newx,int &newy)
{
    newx=xx-yy+n; //不要负数和0
    newy=xx+yy;
}
void update(int xx,int yy,int d)
{
    for(int i=xx; i<=w; i+=lowbit(i))
        for(int j=yy; j<=w; j+=lowbit(j)){
            int pos=lower_bound(h ,h+cnt ,i*w+j)-h;
            c[pos]+=d;
        }
}
int Getsum(int xx,int yy)
{
    int ret=0;
    for(int i=xx; i>0; i-=lowbit(i))
        for(int j=yy; j>0; j-=lowbit(j)){
            int pos=lower_bound(h ,h+cnt ,i*w+j) - h;
            if(h[pos] == i*w+j)   //没有对应的hash值说明这个位置上的c值为0，忽略
                ret+=c[pos];
        }
    return ret;
}
int sum(int x1,int y1,int x2,int y2)
{
    return Getsum(x2,y2)-Getsum(x1-1,y2)-Getsum(x2,y1-1)+Getsum(x1-1,y1-1);
}
int main()
{
    int newx,newy;
    while(scanf("%d",&n) && n){
        w=n*2;
        scanf("%d",&m);
        memset(c,0,sizeof c);
        cnt=0;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d%d",&p[i],&x[i],&y[i],&z[i]);
            if(p[i]==1){
                GetPoint(x[i],y[i],newx,newy);
                Get_ready(newx,newy);
            }
        }
        sort(h,h+cnt);
        for(int i=1;i<=m;i++){
            GetPoint(x[i],y[i],newx,newy);
            if(p[i]==1)
                update(newx,newy,z[i]);
            else{
                int x1,x2,y1,y2;
                x1=max(1,newx-z[i]),x2=min(w,newx+z[i]);
                y1=max(1,newy-z[i]),y2=min(w,newy+z[i]);
                printf("%d\n",sum(x1,y1,x2,y2));
            }
        }
    }
}

转载于:https://www.cnblogs.com/katarinayuan/p/6572868.html