P3810 【模板】三维偏序（陌上花开）

P3810 【模板】三维偏序（陌上花开）

cdq分治+树状数组

三维偏序模板题

前两维用cdq分治，第三维用树状数组进行维护

就像用树状数组搞逆序对那样做--->存权值的出现次数

attention：当两个元素完全相同时要重复计算

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
template <typename T> inline void read(T &x){
    char c=getchar(); x=; bool f=;
    while(!isdigit(c)) f= !f||c=='-' ? :,c=getchar();
    while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=getchar();
    x= f? x:-x;
}int wt[];
template <typename T> inline void output(T x){
    if(!x) {putchar(); return ;}
    if(x<) putchar('-'),x=-x; int l=;
    while(x) wt[++l]=x%,x/=;
    while(l) putchar(wt[l--]+);
}
struct data{
    int x,y,z,ans,w;
    bool operator < (const data &tmp) const{
        return (x<tmp.x||(x==tmp.x&&y<tmp.y)||(x==tmp.x&&y==tmp.y&&z<tmp.z));
    }
}a[],b[];
int n,k,cnt,lev[];
struct tree_array{ //树状数组
    int t[];
    inline void add(int x,int p) {for(;x<=k;x+=x&-x)t[x]+=p;}
    inline int sum(int x) {int res=; for(;x;x-=x&-x)res+=t[x]; return res;}
}mo1;
inline void cdq(int l,int r){ //cdq分治
    int mid=l+((r-l)>>);
    if(l==r) return ;
    cdq(l,mid); cdq(mid+,r);
    int t1=l,t2=mid+;
    for(int i=l;i<=r;++i){
        if((t1<=mid&&a[t1].y<=a[t2].y)||t2>r) mo1.add(a[t1].z,a[t1].w),b[i]=a[t1++]; //注意可以取到等号
        else a[t2].ans+=mo1.sum(a[t2].z),b[i]=a[t2++];
    }
    for(int i=l;i<=mid;++i) mo1.add(a[i].z,-a[i].w); //记得清空树状数组
    for(int i=l;i<=r;++i) a[i]=b[i];
}
int main(){
    read(n); read(k);
    for(int i=;i<=n;++i) read(a[i].x),read(a[i].y),read(a[i].z),a[i].w=;
    sort(a+,a+n+);
    for(int i=;i<=n;++i){
        if(i==||a[i-]<a[i]) a[++cnt]=a[i];
        else ++a[cnt].w; //去重（当然后面要加回来）
    }cdq(,cnt);
    for(int i=;i<=cnt;++i) lev[a[i].ans+a[i].w]+=a[i].w; //把重复的加回来
    for(int i=;i<=n;++i) output(lev[i]),putchar('\n');
    return ;
}