CDQ套CDQ求解三维偏序

陌上花开

---

陌上花开，可缓缓归矣  ——  吴越王

  我们知道CDQ分治套树状数组可以求解三维偏序问题，那么既然CDQ分治能求得降维的问题，如果要用CDQ分治套CDQ分治来解决三维偏序问题呢？

  我们其实就是在CDQ分治直接求解二维偏序上面套了个CDQ分治来解决第一维的问题，并且对第二维进行一个处理。然后，我们不能直接对原数组进行处理，因为那样子会改变他的原有的序号，不再是对第一、二维的直接按序排序了，所以，这里我开了个新的"b[ ]"来进行内层CDQ分治，并且给他加上了原位置的对应的ID，是为了返回答案的时候更为方便了。

说一下操作的流程：

• 首先处理的是外层CDQ分治，我们从递归的底部向上返回值，左区间对右区间产生贡献；
• 利用内层CDQ分治来解决贡献的计算，这时候需要开一个新的数组来存取答案，并且返回到相应位置上去；
• 内层递归，外层递归，直到结束。

样例：

3 5
1 2 1
3 1 2
2 1 2

My Code：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 1e5 + 7;
int N, tot, K, ans[maxN];
struct node
{
    int x, y, z, sum, val, op, id;   //sum是相同元素的个数
    node(int a=0, int b=0, int c=0, int d=0, int f=0):x(a), y(b), z(c), sum(d), val(f) {}
    friend bool operator < (node e1, node e2) { return e1.x == e2.x ? (e1.y == e2.y ? e1.z < e2.z : e1.y < e2.y) : e1.x < e2.x; }
    friend bool operator == (node e1, node e2) { return e1.x == e2.x && e1.y == e2.y && e1.z == e2.z; }
}q[maxN], cop[maxN], b[maxN];
void CDQ_In(int l, int r)
{
    if(l == r) return;
    int mid = HalF;
    CDQ_In(l, mid); CDQ_In(mid + 1, r);
    int ql = l, qr = mid + 1, pos = l, cnt = 0;
    while(ql <= mid || qr <= r)
    {
        if(qr > r || (ql <= mid && b[ql].z <= b[qr].z))
        {
            if(!b[ql].op) cnt += b[ql].sum;
            cop[pos++] = b[ql++];
        }
        else
        {
            if(b[qr].op) q[b[qr].id].val += cnt;
            cop[pos++] = b[qr++];
        }
    }
    for(int i=l; i<=r; i++) b[i] = cop[i];
}
void cdq(int l, int r)
{
    if(l == r) return;
    int mid = HalF;
    cdq(l, mid); cdq(mid + 1, r);
    int ql = l, qr = mid + 1, pos = l;
    while(ql <= mid || qr <= r)
    {
        if(qr > r || (ql <= mid && q[ql].y <= q[qr].y)) { q[ql].op = 0; cop[pos++] = q[ql++]; }
        else { q[qr].op = 1; cop[pos++] = q[qr++]; }
    }
    for(int i=l; i<=r; i++) q[i] = cop[i];
    for(int i=l; i<=r; i++) { b[i] = q[i]; b[i].id = i; }
    CDQ_In(l, r);
}
int main()
{
    scanf("%d%d", &N, &K);
    for(int i=1; i<=N; i++)
    {
        scanf("%d%d%d", &cop[i].x, &cop[i].y, &cop[i].z);
        cop[i].sum = 0;
    }
    sort(cop + 1, cop + N + 1);
    tot = 1;
    q[1] = cop[1];
    int same_one = 0;
    for(int i=2; i<=N; i++)
    {
        if(cop[i] == cop[i-1]) { same_one++; continue; }
        q[tot].sum = same_one + 1;
        q[++tot] = cop[i];
        same_one = 0;
    }
    q[tot].sum = same_one + 1;
    cdq(1, tot);
    for(int i=1; i<=tot; i++)
    {
        ans[q[i].val + q[i].sum - 1] += q[i].sum;
    }
    for(int i=0; i<N; i++) printf("%d\n", ans[i]);
    return 0;
}