当前位置: 代码迷 >> 综合 >> BZOJ 3262: 陌上花开
  详细解决方案

BZOJ 3262: 陌上花开

热度:59   发布时间:2023-12-01 22:00:21.0

经典CDQ分治题,求三维偏序

#include<cstdio>
#include<algorithm>
using namespace std;const int maxn=200000+200;struct Node{
    int a,b,c;int num,level;
}node[maxn];bool cmp(Node A,Node B){
    if(A.a==B.a&&A.b==B.b) return A.c<B.c;if(A.a==B.a) return A.b<B.b;return A.a<B.a; 
}bool cmpcdq(Node A,Node B){
    if(A.b==B.b) return A.c<B.c;return A.b<B.b;
}int tree[maxn],color[maxn];
int ans[maxn];
int type;
int n,k;
int tot;inline int lowbit(int x){
    return x&(-x);
} void Add(int ind,int v){
    while(ind<=k){
    if(color[ind]!=type) tree[ind]=0;color[ind]=type;tree[ind]+=v;ind+=lowbit(ind);}
}int getSum(int ind){
    int sum=0;while(ind>0){
    if(color[ind]==type) sum+=tree[ind];ind-=lowbit(ind);}return sum;
}void CDQ(int l,int r){
    if(l==r) return ;int mid=(l+r)>>1;CDQ(l,mid);CDQ(mid+1,r);sort(node+l,node+mid+1,cmpcdq);sort(node+mid+1,node+r+1,cmpcdq);int i=l;int j=mid+1;type++;for(;j<=r;j++){
    for(;node[i].b<=node[j].b&&i<=mid;i++) Add(node[i].c,node[i].num);node[j].level+=getSum(node[j].c); }
}int main(){
    scanf("%d%d",&n,&k);for(int i=1;i<=n;i++){
    int a,b,c;scanf("%d%d%d",&a,&b,&c);node[i]=Node{
    a,b,c,1,0};}sort(node+1,node+n+1,cmp);tot=1;for(int i=2;i<=n;i++){
    if(node[i].a==node[tot].a&&node[i].b==node[tot].b&&node[i].c==node[tot].c) node[tot].num++;else node[++tot]=node[i];}CDQ(1,tot);for(int i=1;i<=tot;i++) ans[node[i].num+node[i].level-1]+=node[i].num;for(int i=0;i<n;i++) printf("%d\n",ans[i]);
}