[bzoj4822][Cqoi2017]老C的任务&[bzoj1935][Shoi2007]Tree 园丁的烦恼

来自FallDream的博客，未经允许，请勿转载，谢谢。

---

老 C 是个程序员。    

最近老 C 从老板那里接到了一个任务——给城市中的手机基站写个管理系统。作为经验丰富的程序员,老 C 轻松地完成了系统的大部分功能,并把其中一个功能交给你来实现。由于一个基站的面积相对于整个城市面积来说非常的小,因此每个的基站都可以看作坐标系中的一个点,其位置可以用坐标(x, y)来表示。此外,每个基站还有很多属性,例如高度、功率等。运营商经常会划定一个区域,并查询区域中所有基站的信息。现在你需要实现的功能就是,对于一个给定的矩形区域,回答该区域中(包括区域边界上的)所有基站的功率总和。如果区域中没有任何基站,则回答 0。

n,m不知道多大 反正不会太大 不会超过500000吧  然后权值和坐标应该是int

 

因为没有强制在线，所以考虑离线，按照x坐标排序，y坐标离散之后插入线段树。

询问可以拆成两个，差分一下即可。

复杂度nlogn

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
#define MN 1000000
#define N 524288
using namespace std;
inline int read()
{
    int x =  , f = ; char ch = getchar();
    while(ch < '' || ch > ''){ if(ch == '-') f = -;  ch = getchar();}
    while(ch >= '' && ch <= ''){x = x *  + ch - '';ch = getchar();}
    return x * f;
}
ll Ans[MN+],t[N*+];
int cm=,n,m,cnt=,Y[MN+];
struct Tree{int x,y,z;}T[MN+];
struct ques{int x,l,r,id,ad;}q[MN*+];
bool cmp(ques x,ques y){return x.x<y.x;}
bool cmp2(Tree x,Tree y){return x.x<y.x;}
inline void Renew(int x,int ad){for(t[x+=N]+=ad,x>>=;x;x>>=)t[x]=t[x<<]+t[x<<|];}
inline ll Query(int l,int r)
{
    if(l>r) return ;ll sum=;
    for(l+=N-,r+=N+;l^r^;l>>=,r>>=)
    {
        if(~l&) sum+=t[l+];
        if( r&) sum+=t[r-];
    }
    return sum;
}
int main()
{
    n=read();m=read();
    for(int i=;i<=n;++i) T[i].x=read(),Y[i]=T[i].y=read(),T[i].z=read();
    sort(Y+,Y+n+);
    for(int i=;i<=n;++i)  if(Y[i]!=Y[i-]) Y[++cm]=Y[i];
    for(int i=;i<=n;++i)
        T[i].y=lower_bound(Y+,Y+cm+,T[i].y)-Y;
    for(int i=;i<=m;++i)
    {
        int X1=read(),Y1=read(),X2=read(),Y2=read();
        Y1=lower_bound(Y+,Y+cm+,Y1)-Y;
        Y2=upper_bound(Y+,Y+cm+,Y2)-Y-;
        q[++cnt]=(ques){X1-,Y1,Y2,i,-};
        q[++cnt]=(ques){X2,Y1,Y2,i,};
    }
    sort(q+,q+cnt+,cmp);
    sort(T+,T+n+,cmp2);
    for(int i=,j=;j<=cnt;)
        if(i<=n&&T[i].x<=q[j].x) Renew(T[i].y,T[i].z),++i;
        else Ans[q[j].id]+=q[j].ad*Query(q[j].l,q[j].r),++j;
    for(int i=;i<=m;++i) printf("%lld\n",Ans[i]);
    return ;
}