BZOJ 3262 陌上花开 CDQ分治

= =原来复杂度还是nlog^2(n) Orz 被喷了

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;

const int D=3e6;
char in[D],*I=in,out[D],*O=out;

inline int gint(){
    int x=0;for(;*I<48||*I>57;++I);
    for(;*I>47&&*I<58;++I)x=10*x+*I-48;
    return x;
}

inline void print(int x){
    char tmp[10],*t=tmp;
    if(!x)*t++=48;
    for(;x;x/=10)*t++=x%10+48;
    for(;t--!=tmp;*O++=*t);
    *O++='
';
}

const int Maxn=100000 + 10;
int C[Maxn*2],ans[Maxn],N,K,n=0;
struct Node{
    int a,b,c,s,ans;
    inline bool operator==(const Node&rhs)const{
        return a==rhs.a&&b==rhs.b&&c==rhs.c;
    }
    inline bool operator!=(const Node&rhs)const{
        return !(*this==rhs);
    }
    void init(){a=gint(),b=gint(),c=gint();}
}p[Maxn],q[Maxn];

inline void Add(int x,const int&y){
    for(;0<x&&x<=K;x+=x&-x)C[x]+=y;
}
inline int Query(int x){
    int ret=0;
    for(;0<x&&x<=K;x-=x&-x)ret+=C[x];
    return ret;
}
inline bool cmp1(const Node&x,const Node&y){
    if(x.a!=y.a)return x.a<y.a;
    if(x.b!=y.b)return x.b<y.b;
    return x.c<y.c;
}
inline bool cmp2(const Node&x,const Node&y){
    if(x.b!=y.b)return x.b<y.b;
    return x.c<y.c; 
}
void CDQ(int l,int r){
    if(l==r)return;
    int mid=(l+r)>>1;
    CDQ(l,mid);CDQ(mid+1,r);
    
    int i=l;
    for(int j=mid+1;j<=r;j++){
        for(;i<=mid&&p[i].b<=p[j].b;i++)
            Add(p[i].c,p[i].s);
        p[j].ans+=Query(p[j].c);
    }
    for(int j=l;j<i;j++)Add(p[j].c,-p[j].s);
    merge(p+l,p+mid+1,p+mid+1,p+r+1,q,cmp2);
    memcpy(p+l,q,sizeof(p[0])*(r-l+1));
}
void init(){
    N=gint(),K=gint();
    for(int i=1;i<=N;i++)q[i].init();
    sort(q+1,q+N+1,cmp1);
    for(int cnt=1,i=1;i<=N;i++,cnt++){
        if(q[i]!=q[i+1]){p[++n]=q[i];p[n].s=cnt;cnt=0;}
    }
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    
    fread(I,1,D,stdin);
    
    init();
    CDQ(1,n);
    
    for(int i=1;i<=n;i++)ans[p[i].ans+p[i].s-1]+=p[i].s;
    for(int i=0;i<N;i++)print(ans[i]);
    
    *--O=0;
    
    return puts(out),0;
}