【bzoj1818】 CQOI2010内部白点 树状数组计数

真是道好题，题解在这里http://hzwer.com/?s=1818。

说一下WA点吧，加线段的时候要分开加，不能一列只加一个，因为会把中间的初始黑点统计到。

一定是统计第一秒中染的黑点+初始黑点。

cmp函数比较难写。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<iostream>#include<algorithm>#include<map>#define maxn 100010using namespace std;struct yts{int x,y;}a[maxn];struct yts1{int typ,y1,x1,x2;}h[1000010];map<int,int> p1,p2;int n,m,cnt,ans;int t[maxn];int c[maxn];bool cmp1(yts x,yts y){return x.x<y.x || (x.x==y.x && x.y<y.y);}bool cmp2(yts x,yts y){return x.y<y.y || (x.y==y.y && x.x<y.x);}bool cmp3(yts1 x,yts1 y){return x.y1<y.y1 || (x.y1==y.y1 && x.typ<y.typ);}void addedge(int f,int x,int y,int z)//1横0竖 {if (f==1){h[++cnt].typ=0;h[cnt].y1=x;h[cnt].x1=y;h[cnt].x2=z;}else{h[++cnt].typ=1;h[cnt].y1=y;h[cnt].x1=x;h[++cnt].typ=-1;h[cnt].y1=z;h[cnt].x1=x;}}int lowbit(int x){return x&(-x);}int query(int i){int ans=0;while (i){ans+=c[i];i-=lowbit(i);}return ans;}void add(int i,int x){while (i<=n){c[i]+=x;i+=lowbit(i);}}int main(){scanf("%d",&n);for (int i=1;i<=n;i++) scanf("%d%d",&a[i].x,&a[i].y);for (int i=1;i<=n;i++) t[i]=a[i].x;sort(t+1,t+n+1);int mx=1;p1[t[1]]=1;for (int i=2;i<=n;i++) if (t[i]!=t[i-1]) p1[t[i]]=++mx;for (int i=1;i<=n;i++) t[i]=a[i].y;sort(t+1,t+n+1);mx=1;p2[t[1]]=1;for (int i=2;i<=n;i++) if (t[i]!=t[i-1]) p2[t[i]]=++mx;for (int i=1;i<=n;i++) a[i].x=p1[a[i].x],a[i].y=p2[a[i].y];sort(a+1,a+n+1,cmp1);for (int i=2;i<=n;i++)if (a[i].x==a[i-1].x) addedge(0,a[i].x,a[i-1].y,a[i].y);sort(a+1,a+n+1,cmp2);for (int i=2;i<=n;i++)if (a[i].y==a[i-1].y) addedge(1,a[i].y,a[i-1].x,a[i].x);sort(h+1,h+cnt+1,cmp3);for (int i=1;i<=cnt;i++){  if (!h[i].typ) ans+=query(h[i].x2-1)-query(h[i].x1);  else add(h[i].x1,h[i].typ);}printf("%d\n",ans+n);return 0;}