【树状数组】bzoj1935 [Shoi2007]Tree 园丁的烦恼

把y坐标离散化后,按x坐标排序,把询问拆成四个点,每次询问某个点左下角的点的个数,注意处理边界和重叠的情况。

#include<cstdio>
#include<algorithm>
using namespace std;
#define N 500001
int f,C;
void R(int &x){
    C=0;f=1;
    for(;C<'0'||C>'9';C=getchar())if(C=='-')f=-1;
    for(x=0;C>='0'&&C<='9';C=getchar())(x*=10)+=(C-'0');
    x*=f;
}
void P(int x){
    if(x<10)putchar(x+'0');
    else{P(x/10);putchar(x%10+'0');}
}
struct LiSan{int p,v;}t[N*5];
bool operator < (const LiSan &a,const LiSan &b){return a.v<b.v;}
struct Point{int x,y,type,id;}p[N*5];
bool operator < (const Point &a,const Point &b)
{
    if(a.x!=b.x) return a.x<b.x;
    if(a.y!=b.y) return a.y<b.y;
    return a.type<b.type;
}
int n,m,en,zy;
int d[N*5],anss[N];
void Update(int p){for(;p<=zy;p+=(p&(-p))) ++d[p];}
int Query(int p){int res=0; for(;p;p-=(p&(-p))) res+=d[p]; return res;}
int main()
{
    R(n); R(m);
    for(int i=1;i<=n;++i)
      {
        R(p[i].x); R(t[i].v);
        t[i].p=i;
      }
    en=n;
    for(int i=1;i<=m;++i)
      {
        ++en; R(p[en].x); R(t[en].v); t[en].p=en; p[en].type=1; p[en].id=i;
        ++en; R(p[en+2].x); R(t[en+2].v); p[en].id=i;
        p[en].x=p[en-1].x; t[en].v=t[en+2].v; p[en].type=2; t[en].p=en;
        ++en; p[en].x=p[en+1].x; t[en].v=t[en-2].v; t[en].p=en; p[en].type=3; p[en].id=i;
        ++en; p[en].type=4; p[en].id=i; t[en].p=en;
        --p[en-3].x; --t[en-3].v;
        --p[en-2].x;
        --t[en-1].v;
      }
    sort(t+1,t+en+1);
    p[t[1].p].y=++zy;
    for(int i=2;i<=en;++i)
      {
        if(t[i].v!=t[i-1].v) ++zy;
        p[t[i].p].y=zy; 
      }
    sort(p+1,p+en+1);
    for(int i=1;i<=en;++i)
      if(!p[i].type) Update(p[i].y);
      else if(p[i].type==1||p[i].type==4) anss[p[i].id]+=Query(p[i].y);
      else anss[p[i].id]-=Query(p[i].y);
    for(int i=1;i<=m;++i) P(anss[i]),puts("");
    return 0;
}
原文地址:https://www.cnblogs.com/autsky-jadek/p/4444367.html