BZOJ3262: 陌上花开(cdq分治)

#include<cstdio>
#include<algorithm>
#define lowbit(x) ((x) & (-x))
using namespace std;
const int MAXN = 200001;
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int N, Max;
struct Node {
    int a, b, c, f, w;
    bool operator < (const Node &rhs) const{
        return (a == rhs.a && b == rhs.b) ? (c < rhs.c) : (a == rhs.a ? b < rhs.b : a < rhs.a);
    }
}A[MAXN], tp[MAXN];
int num = 0, ans[MAXN];
int T[MAXN];
void Add(int pos, int val) {
    for(int i = pos; i <= Max; i += lowbit(i)) T[i] += val;
}
int Query(int pos) {
    int ans = 0;
    for(int i = pos; i; i -= lowbit(i)) ans += T[i];
    return ans;
}
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, j = mid + 1, P = l;
    while(i <= mid || j <= r) {
        if(j > r || (i <= mid && A[i].b <= A[j].b)) Add(A[i].c, A[i].w), tp[P++] = A[i++];
        else A[j].f += Query(A[j].c), tp[P++] = A[j++];
    }
    for(int i = l; i <= mid; i++) Add(A[i].c, -A[i].w);
    for(int i = l; i <= r; i++) A[i] = tp[i];
}
int main() {
#ifdef WIN32
    freopen("a.in", "r", stdin);
#endif
    N = read(); Max = read();
    for(int i = 1; i <= N; i++) A[i].a = read(), A[i].b = read(), A[i].c = read(), A[i].w = 1;
    sort(A + 1, A + N + 1);
    num = 1;
    for(int i = 2; i <= N; i++)
        if(A[i].a == A[num].a && A[i].b == A[num].b && A[i].c == A[num].c) A[num].w++;
        else A[++num] = A[i];
    CDQ(1, num);
    for(int i = 1; i <= num; i++) ans[A[i].f + A[i].w - 1] += A[i].w;
    for(int i = 0; i <= N - 1; i++)
        printf("%d\n", ans[i]);
    return 0;
}