重庆OI2017 老 C 的任务

老 C 的任务

时间限制: 2 Sec 内存限制: 512 MB

题目描述

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

输入

第一行两个整数n, m ，表示一共有n个基站和m 次查询。
接下来一共有n行，每行由xB,yB,pB三个空格隔开的整数构成，表示一个基站的坐标(xB,yB)和功率pB。不会有两个基站位于同一坐标。
接下来一共有m行，每行由x1E,y1E,x2E,y2E四个空格隔开的整数构成，表示一次查询的矩形区域。该矩形对角坐标为(x1E,y1E)和(x2E,y2E)，且 4 边与坐标轴平行。

输出

输出m行，每行一个整数，对应每次查询的结果。

样例输入

样例输出

11

4

提示

1≤n≤100000,1≤m≤100000。-231≤xi,yi,pi,x1j,y1j,x2j,y2j<231,x1j≤x2j,y1j≤y2j

分析：求矩形面积，离线扫描线，树状数组或线段树维护前缀和即可；

代码：

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <cmath>

#include <algorithm>

#include <climits>

#include <cstring>

#include <string>

#include <set>

#include <bitset>

#include <map>

#include <queue>

#include <stack>

#include <vector>

#include <cassert>

#include <ctime>

#define rep(i,m,n) for(i=m;i<=(int)n;i++)

#define mod 1000000007

#define inf 0x3f3f3f3f

#define vi vector<int>

#define pb push_back

#define mp make_pair

#define fi first

#define se second

#define ll long long

#define pi acos(-1.0)

#define pii pair<int,int>

#define sys system("pause")

#define ls rt<<1

#define rs rt<<1|1

#define all(x) x.begin(),x.end()

const int maxn=3e5+;

const int N=5e2+;

using namespace std;

ll gcd(ll p,ll q){return q==?p:gcd(q,p%q);}

ll qpow(ll p,ll q,ll mo){ll f=;while(q){if(q&)f=f*p%mo;p=p*p%mo;q>>=;}return f;}

int n,m,k,t,x[maxn<<],y[maxn<<];

ll sum[maxn<<],ret[maxn];

struct node

{

    int x,y,z;

}a[maxn];

struct query

{

    node l,r;

}q[maxn];

vi qu[maxn];

vector<pii>pos[maxn];

void pup(int rt)

{

    sum[rt]=sum[ls]+sum[rs];

}

void upd(int L,ll v,int l,int r,int rt)

{

    if(l==r)

    {

        sum[rt]+=v;

        return;

    }

    int mid=l+r>>;

    if(L<=mid)upd(L,v,l,mid,ls);

    else upd(L,v,mid+,r,rs);

    pup(rt);

}

ll gao(int L,int R,int l,int r,int rt)

{

    if(L==l&&R==r)

    {

        return sum[rt];

    }

    int mid=l+r>>;

    if(R<=mid)return gao(L,R,l,mid,ls);

    else if(L>mid)return gao(L,R,mid+,r,rs);

    else return gao(L,mid,l,mid,ls)+gao(mid+,R,mid+,r,rs);

}

int main()

{

    int i,j;

    scanf("%d%d",&n,&m);

    rep(i,,n)

    {

        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);

        x[++x[]]=a[i].x,y[++y[]]=a[i].y;

    }

    rep(i,,m)

    {

        scanf("%d%d%d%d",&q[i].l.x,&q[i].l.y,&q[i].r.x,&q[i].r.y);

        x[++x[]]=q[i].l.x,x[++x[]]=q[i].r.x;

        y[++y[]]=q[i].l.y,y[++y[]]=q[i].r.y;

    }

    sort(x+,x+x[]+);

    x[]=unique(x+,x+x[]+)-x-;

    sort(y+,y+y[]+);

    y[]=unique(y+,y+y[]+)-y-;

    rep(i,,n)

    {

        a[i].x=lower_bound(x+,x+x[]+,a[i].x)-x;

        a[i].y=lower_bound(y+,y+y[]+,a[i].y)-y;

        pos[a[i].x].pb(mp(a[i].y,a[i].z));

    }

    rep(i,,m)

    {

        q[i].l.x=lower_bound(x+,x+x[]+,q[i].l.x)-x;

        q[i].l.y=lower_bound(y+,y+y[]+,q[i].l.y)-y;

        q[i].r.x=lower_bound(x+,x+x[]+,q[i].r.x)-x;

        q[i].r.y=lower_bound(y+,y+y[]+,q[i].r.y)-y;

        qu[q[i].l.x-].pb(i);

        qu[q[i].r.x].pb(i);

    }

    rep(i,,x[])

    {

        rep(j,,pos[i].size()-)upd(pos[i][j].fi,pos[i][j].se,,y[],);

        rep(j,,qu[i].size()-)

        {

            if(i==q[qu[i][j]].l.x-)ret[qu[i][j]]-=gao(q[qu[i][j]].l.y,q[qu[i][j]].r.y,,y[],);

            else ret[qu[i][j]]+=gao(q[qu[i][j]].l.y,q[qu[i][j]].r.y,,y[],);

        }

    }

    rep(i,,m)printf("%lld\n",ret[i]);

    return ;

}