Tree 园丁的烦恼 bzoj-1935 Shoi-2007
题目大意:给定平面上的$n$个点,$m$次查询矩形点个数。
注释:$1le n,mle 5cdot 10^5$。
想法:静态二维数点。
$Orz Winniechen$,真tm敢写$KD-Tree$,虽然$T$了..
正常这种静态的二维数点我们都要请到树状数组。最简单的就是二维树状数组。
但是发现开不下,这样的话我们依据它可以离线这一点,我们将每个询问$(x1,y1)$到$(x2,y2)$变成$4$次查询:
$(x1-1,y1-1),(x1-1,y2),(x2,y1-1),(x2,y2)$把它们哥四个都当成点放入点集。每次都相当于查询$(1,1)$到$balabala$
每个点集中的点有$4$个参数:横纵坐标,种类和系数。
种类就是这个点到底是给定的点还是查询的点。
系数的话就是容斥前面的系数:$(x1-1,y1-1)$和$(x2,y2)$前面是$1$,$(x1-1,y2)$和$(x2,y1-1)$前面是$-1$。
然后我们将点集排序,横坐标递增为第一关键字,纵坐标递增为第二关键字。
紧接着我们顺次枚举每个点,如果这个点的种类是给定点,我们将它压到树状数组里。
是一个树状数组里,我们只开一个树状数组,记录的是小于每个横坐标的点的个数。
这样的话根据我们的关键字可知,每一个有可能更新查询的点都会被提前枚举过。
如果这个点是查询,就直接查询然后累加到对应查询的编号答案上,即可。
最后,附上丑陋的代码... ...
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #define N 1000010 using namespace std; struct Node { int x,y,f,id;//这里跟上述的有些不一样。 //我们用f直接可以同时记录种类和系数。如果f==0,那么就表示这个点是给定的点,反之f=1或-1代表系数,直接累计即可。 }q[N<<2]; int tree[10000010]; int mx=0; int ans[N]; inline bool cmp(const Node &a,const Node &b) { return a.x!=b.x?a.x<b.x:(a.y!=b.y?a.y<b.y:a.id<b.id); } int a,b,c,d; int cnt; inline int lowbit(int x) {return x&(-x);} void fix(int x) { // if(!x) x++; // puts("fix"); for(int i=x;i<=mx+1;i+=lowbit(i)) { // printf("aha %d ",i); tree[i]++; } } int query(int x) { // puts("query"); int ans=0; for(int i=x;i>=1;i-=lowbit(i)) { ans+=tree[i]; } return ans; } int main() { int n,m; cin >> n >> m ; for(int i=1;i<=n;i++) { scanf("%d%d",&q[i].x,&q[i].y); q[i].x++,q[i].y++; mx=max(mx,q[i].y); } cnt=n; // puts("Fuck 1"); for(int i=1;i<=m;i++) { scanf("%d%d%d%d",&a,&b,&c,&d); c++,d++; q[++cnt].x=c; q[cnt].y=d; q[cnt].f=1; q[cnt].id=i;//这里不要把系数搞错 q[++cnt].x=c; q[cnt].y=b; q[cnt].f=-1; q[cnt].id=i; q[++cnt].x=a; q[cnt].y=d; q[cnt].f=-1; q[cnt].id=i; q[++cnt].x=a; q[cnt].y=b; q[cnt].f=1; q[cnt].id=i; } // printf("Gun %d ",cnt); // puts("Fuck 2"); sort(q+1,q+cnt+1,cmp); for(int i=1;i<=cnt;i++) { // printf("Shit %d ",i); if(!q[i].id) fix(q[i].y); else ans[q[i].id]+=query(q[i].y)*q[i].f;//这就是直接压到一个变量的好处,不用特判直接加就行了,反正如果是给定点f就是0,不会影响答案。 } // puts("Fuck 3"); for(int i=1;i<=m;i++) { printf("%d ",ans[i]); } return 0; }
小结:这个想法极其常用,很多静态可离线二维数点问题都可以用这个来解决。那些动态的就让KD-Tree上吧,反正出了那种题被卡常的不可能只有你一个人/手动滑稽