3262: 陌上花开

3262: 陌上花开

链接

分析

　　三维偏序问题。CDQ分治+树状数组。

　　首先没注意如何输出，就只直接把f[i]输出了，调啊调......然后没注意是小于等于又调啊调......最后没有特判相同的，看了题解才知道......

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
 
inline int read() {
    int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for (;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f;
}
 
const int N = 100100;
int ans[N],d[N],n,m;
struct Node{
    int x,y,z,cnt,ans;
    bool operator < (const Node &a) const {
        if (x == a.x && y == a.y) return z < a.z;
        if (x == a.x) return y < a.y;
        return x < a.x; 
    }
}A[N],B[N];
struct BIT{
    int sum[N<<1];
    void update(int p,int v) {
        for (; p<=m; p+=p&(-p)) sum[p] += v;
    }
    int query(int p) {
        int ans = 0;
        for (; p; p-=p&(-p)) ans += sum[p];
        return ans;
    }
    int clear(int p) {
        for (; p<=m&&sum[p]; p+=p&(-p)) sum[p] = 0;
    }
}bit;
 
void CDQ(int L,int R) {
    if (L == R) {A[L].ans += A[L].cnt - 1;return ;}
    int M = (L + R) >> 1;
    CDQ(L,M);
    CDQ(M+1,R);
    int i = L,j = M + 1,k = L;
    while (i <= M && j <= R) {
        if (A[i].y <= A[j].y ) { // 对y进行排序，y相同时，z最初已经排好了。 
            bit.update(A[i].z,A[i].cnt);
            B[k++] = A[i++];
        }
        else {
            A[j].ans += bit.query(A[j].z);
            B[k++] = A[j++];
        }
    }
    while (i <= M) B[k++] = A[i++];
    while (j <= R) {
        A[j].ans += bit.query(A[j].z);
        B[k++] = A[j++];
    }
    for (i=L; i<=R; ++i) {
        bit.clear(A[i].z);
        A[i] = B[i];
    }
}
int main() {
    n = read(),m = read(); 
    for (int i=1; i<=n; ++i) {
        A[i].x = read(),A[i].y = read(),A[i].z = read(),A[i].cnt = 1;
    }
    sort(A+1,A+n+1);
    int tot = 1;
    for (int i=2; i<=n; ++i) { // 去除相同的元素。 
        if (A[i].x == A[tot].x && A[i].y == A[tot].y && A[i].z == A[tot].z) A[tot].cnt ++;
        else A[++tot] = A[i];
    }
     
    CDQ(1,tot);
    for (int i=1; i<=tot; ++i) ans[A[i].ans] += A[i].cnt;
    for (int i=0; i<n; ++i) printf("%d
",ans[i]);
    return 0;
}