[掃描線-線段樹] [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;
}