[扫描线-线段树] [CQOI2017]Day2 T1 老C的任务

题目描述

传送门

老C是个程序员。

最近老C从老板那里接到了一个任务——给城市中的手机基站写个管理系统。作为经验丰富的程序员，老C轻松地完成了系统的大部分功能，并把其中一个功能交给你来实现。

由于一个基站的面积相对于整个城市面积来说非常的小，因此每个的基站都可以看作坐标系中的一个点，其位置可以用坐标(x,y)来表示。此外，每个基站还有很多属性，例如高度、功率等。运营商经常会划定一个区域，并查询区域中所有基站的信息。

现在你需要实现的功能就是，对于一个给定的矩形区域，回答该区域中（包括区域边界上的）所有基站的功率总和。如果区域中没有任何基站，则回答0。

分析

考虑扫描线

理解离散化函数的含义：
unique去重后返回无重数组的尾指针
lower_bound(a.begin,a.end,k,cmp)返回值严格大于k，减数组之后得到指针

从左往右扫完事
扫的时候只能从左到右连续扫，那么处理前缀和得答案的方法就是扫到那根线的时候处理（+或-）

Code

欲让RE爬，先把数组开大

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define in Read()
#define re register
#define mid (tr[p].l+tr[p].r)>>1
inline int in{
	int i=0,f=1;char ch=getchar();
	while(!isdigit(ch)&&ch!='-')ch=getchar();
	if(ch=='-')ch=getchar(),f=-1;
	while(isdigit(ch))i=(i<<1)+(i<<3)+ch-48,ch=getchar();
	return i*f;
}

const int NNN=5e5+10;
int n,m;
int disc[NNN<<2],lim;//离散化 
struct Point{
	int x,y,wei;
}p[NNN];
struct Query{
	int x,l,r,id,opt;
}q[NNN<<1];
int ans[NNN];
struct Tree{
	int l,r,val;
}tr[NNN<<2];

inline bool cmp1(const Point &u,const Point &v){return u.x<v.x;}
inline bool cmp2(const Query &u,const Query &v){return u.x<v.x;}
inline void push_up(int p){tr[p].val=tr[p<<1].val+tr[p<<1|1].val;}

inline void build(int p,int l,int r){
	tr[p].l=l,tr[p].r=r;
	if(l==r)return;
	int midd=(l+r)>>1;
	build(p<<1,l,midd);
	build(p<<1|1,midd+1,r);
}

inline void update(int p,int k,int w){
	if(tr[p].l==tr[p].r){
		tr[p].val+=w;
		return;
	}
	
	if(k<=mid)update(p<<1,k,w);
	else update(p<<1|1,k,w);
	push_up(p);
	return;
}

inline int query(int p,int L,int R){
	if(L<=tr[p].l&&tr[p].r<=R)return tr[p].val;
	
	int res=0;
	if(L<=mid)res+=query(p<<1,L,R);
	if(R> mid)res+=query(p<<1|1,L,R);
	return res;
}

signed main(){
	n=in,m=in;
	
	for(re int i=1;i<=n;++i){
		p[i].x=in;
		disc[++lim]=p[i].y=in;
		p[i].wei=in;
	}
	
	for(re int i=1;i<=(m<<1);i+=2){
		q[i].x=in-1;
		disc[++lim]=q[i].l=in;
		q[i+1].x=in;
		disc[++lim]=q[i].r=in;
		
		q[i+1].id=q[i].id=(i>>1)+1;
		q[i].opt=-1;
		q[i+1].opt=1;
	}
	
	sort(disc+1,disc+lim+1);
	lim=unique(disc+1,disc+lim+1)-disc-1;
	
	for(re int i=1;i<=n;++i)
		p[i].y=lower_bound(disc+1,disc+lim+1,p[i].y)-disc;
	
	for(re int i=1;i<=(m<<1);i+=2){
		q[i+1].l=q[i].l=lower_bound(disc+1,disc+lim+1,q[i].l)-disc;
		q[i+1].r=q[i].r=lower_bound(disc+1,disc+lim+1,q[i].r)-disc;
	}
	
	sort(p+1,p+n+1,cmp1);
	sort(q+1,q+(m<<1)+1,cmp2);
	
	build(1,1,lim);
	re int pos=1;
	for(int i=1;i<=(m<<1);++i){
		while(pos<=n&&p[pos].x<=q[i].x){
			update(1,p[pos].y,p[pos].wei);
			++pos;
		}
		ans[q[i].id]+=query(1,q[i].l,q[i].r)*q[i].opt;
	}
	for(re int i=1;i<=m;++i)printf("%lld\n",ans[i]);
	
	return 0;
}