【BZOJ4822】[CQOI2017]老C的任务（扫描线）

【BZOJ4822】[CQOI2017]老C的任务（扫描线）

题面

BZOJ

洛谷

题解

没有修改操作，都不需要分治了。。。

直接排序之后扫描线算贡献就好了。。。

不知道为啥洛谷上过不了。。。

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<vector>

using namespace std;

#define ll long long

#define MAX 100100

inline int read()

{

	int x=0;bool t=false;char ch=getchar();

	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

	if(ch=='-')t=true,ch=getchar();

	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

	return t?-x:x;

}

int n,m;

struct Node{int x,y,p;}p[MAX];

struct Qry{int l,r,x,opt,id;}q[MAX<<1];

bool operator<(Node a,Node b){return a.x<b.x;}

bool operator<(Qry a,Qry b){return a.x<b.x;}

int X[MAX<<2],Y[MAX<<2],tx,ty,tot;

int calcx(int x){return lower_bound(&X[1],&X[tx+1],x)-X;}

int calcy(int x){return lower_bound(&Y[1],&Y[ty+1],x)-Y;}

ll ans[MAX],c[MAX<<2];

int lb(int x){return x&(-x);}

void add(int x,int w){while(x<=ty)c[x]+=w,x+=lb(x);}

ll getsum(int x){ll ret=0;while(x)ret+=c[x],x-=lb(x);return ret;}

int main()

{

	n=read();m=read();

	for(int i=1;i<=n;++i)

	{

		int x=read(),y=read(),P=read();

		X[++tx]=x;Y[++ty]=y;

		p[i]=(Node){x,y,P};

	}

	for(int i=1;i<=m;++i)

	{

		int x1=read(),y1=read(),x2=read(),y2=read();

		q[++tot]=(Qry){y1,y2,x1,-1,i};

		q[++tot]=(Qry){y1,y2,x2,1,i};

		X[++tx]=x1;X[++tx]=x2;Y[++ty]=y1;Y[++ty]=y2;

	}

	sort(&X[1],&X[tx+1]);tx=unique(&X[1],&X[tx+1])-X-1;

	sort(&Y[1],&Y[ty+1]);ty=unique(&Y[1],&Y[ty+1])-Y-1;

	for(int i=1;i<=n;++i)p[i].x=calcx(p[i].x),p[i].y=calcy(p[i].y);

	for(int i=1;i<=tot;++i)q[i].x=calcx(q[i].x),q[i].l=calcy(q[i].l),q[i].r=calcy(q[i].r);

	sort(&p[1],&p[n+1]);sort(&q[1],&q[tot+1]);

	for(int i=1,j=0;i<=tot;++i)

	{

		while(j<n&&p[j+1].x<q[i].x)++j,add(p[j].y,p[j].p);

		if(q[i].opt==-1)ans[q[i].id]-=getsum(q[i].r)-getsum(q[i].l-1);

		while(j<n&&p[j+1].x<=q[i].x)++j,add(p[j].y,p[j].p);

		if(q[i].opt==1)ans[q[i].id]+=getsum(q[i].r)-getsum(q[i].l-1);

	}

	for(int i=1;i<=m;++i)printf("%lld\n",ans[i]);

	return 0;

}