CDQ分治复习(一)三维偏序(P3810陌上开花)

2023大厂真题提交网址(含题解):

www.CodeFun2000.com（http://101.43.147.120/）

最近我们一直在将收集到的机试真题制作数据并搬运到自己的OJ上，供大家免费练习,体会真题难度。现在OJ已录入50+道2023年最新大厂真题，同时在不断的更新。同时，可以关注"塔子哥学算法"公众号获得每道题的题解。

背景:

三维偏序的板子,但是有很多细节要处理.

题目大意:

给你$n$个三元组$(a,b,c)$.问你对于每个三元组$i$,有多少个三元组$j$满足$a_i\le a_j,b_i\le b_j,c_i\le c_j$.

题目思路:

1.把$a$这一维看成是[之前求逆序对时]的下标.要对$a$进行排序.
这样我们在分治的过程中,保证左半边的$a$总是小于等于右半边的$a$.

PS:排序时要考虑$a$相等的情况.
1.若偏序关系没有等号,则需要对第2/3维降序排序。因为$a$相等的这数互相不构成偏序关系，逆着来不会造成点对贡献.
2.若有等号，对第2/3维顺序排序即可.
2.CDQ分治的过程中,我们保证了左边的$b$是升序的且右边的$b$也是升序的.所以可以利用扫描线+树状数组or继续套cdq。对左边的$z$插入,右边查询$z$来计算贡献.最后记得撤销树状数组的贡献.

3.计算完贡献之后再按$b$来合并两数组.(正常归并排序)

坑点:
1.三元组有重复的情况:需要去重,将相同的点绑定在一起分治(具体看代码实现).因为相同的点对之间成环了.要特殊处理.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define mid ((l + r) >> 1)
const int maxn = 1e5 + 5;
const int maxk = 2e5 + 5;
const int mod = 1e9 + 7;
int sum[maxk];
int n , k , ans[maxn] , res[maxn];
int lowbit(int x){return x & -x;}
void add(int x,int c){while(x<=k){sum[x] += c;x += lowbit(x);}}
int ask(int x){int ans=0;while(x){ans+=sum[x];x-=lowbit(x);}return ans;}
struct Node{
    int x , y , z , id;
    bool operator == (const Node & a){
        return x == a.x && y == a.y && z == a.z;
    }
}a[maxn] , b[maxn] , c[maxn];
bool cmp1 (const Node & a , const Node & b){
    if (a.x == b.x){
        if (a.y == b.y) return a.z < b.z;
        return a.y < b.y;
    }
    return a.x < b.x;
}
bool cmp2 (const Node & a , const Node & b){
    return a.y < b.y;
}
// 第i个出现了多少次
int bk[maxn];
void cdq (int l , int r){
    if (l == r) return ;
    cdq(l , mid);
    cdq(mid + 1 , r);

    int now = l;
    for (int i = mid + 1 ; i <= r ; i++){
        while (now <= mid && a[now].y <= a[i].y){
            add(a[now].z , bk[a[now].id]); //bk[a[now].id]
            now++;
        }
        ans[a[i].id] += ask(a[i].z);
    }
    for (int i = l ; i < now ; i++) add(a[i].z , -bk[a[i].id]);

    int x = l , y = mid + 1 , cnt = l;
    while (x <= mid && y <= r){
        if (a[x].y <= a[y].y) b[cnt++] = a[x++];
        else b[cnt++] = a[y++];
    }
    while (x <= mid) b[cnt++] = a[x++];
    while (y <= r) b[cnt++] = a[y++];
    for (int i = l ; i <= r ; i++) a[i] = b[i];
    return ;
}
int main()
{
    scanf("%d%d",&n,&k);
    for (int i = 1 ; i <= n ; i++){
        scanf("%d%d%d" , &c[i].x , &c[i].y , &c[i].z);
    }
    sort(c + 1 , c + 1 + n , cmp1);
    int gg = 0;
    for (int i = 1 ; i <= n ; i++){
        int j = i;
        while (j <= n && c[j + 1] == c[i]) j++;
        a[++gg] = c[i];
        a[gg].id = gg;
        bk[gg] = j - i + 1;
        i = j;
    }
    cdq(1 , gg);
    for (int i = 1 ; i <= gg ; i++){
        ans[i] += bk[i] - 1;
        res[ans[i]] += bk[i];
    }
    for (int i = 0 ; i < n ; i++){
        printf("%d\n" , res[i]);
    }
    return 0;
}