洛谷 P3810 【模板】三维偏序（陌上花开）cdq分治

题目背景

这是一道模板题

可以使用bitset，CDQ分治，K-DTree等方式解决。

题目描述

有 n n 个元素，第 i i 个元素有 a_i a
i
​ 、 b_i b
i
​ 、 c_i c
i
​ 三个属性，设 f(i) f(i) 表示满足 a_j \leq a_i a
j
​ ≤a
i
​ 且 b_j \leq b_i b
j
​ ≤b
i
​ 且 c_j \leq c_i c
j
​ ≤c
i
​ 的 j j 的数量。

对于 d \in [0, n) d∈[0,n) ，求 f(i) = d f(i)=d 的数量

输入输出格式

输入格式：
第一行两个整数 n n 、 k k ，分别表示元素数量和最大属性值。

之后 n n 行，每行三个整数 a_i a
i
​ 、 b_i b
i
​ 、 c_i c
i
​ ，分别表示三个属性值。

输出格式：
输出 n n 行，第 d + 1 d+1 行表示 f(i) = d f(i)=d 的 i i 的数量。

输入输出样例

输入样例#1： 复制
10 3
3 3 3
2 3 3
2 3 1
3 1 1
3 1 2
1 3 1
1 1 2
1 2 2
1 3 2
1 2 1
输出样例#1： 复制
3
1
3
0
1
0
1
0
0
1
说明
1≤n≤100000,1≤k≤200000

分析：一道模版题。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>

const int maxn=1e5+7;

using namespace std;

int n,k;
int t[maxn*5],f[maxn];

struct node{
    int x,y,z,ans;
}a[maxn];

struct rec{
    int y,z,t,c;
}b[maxn];

bool cmp(node x,node y)
{
    if (x.x==y.x)
    {
        if (x.y==y.y) return x.z<y.z;
        return x.y<y.y;
    }
    return x.x<y.x;
}

int lowbit(int x)
{
    return x&(-x);
}

void updata(int x,int c)
{
    while (x<=k)
    {
        t[x]+=c;
        x+=lowbit(x);
    }
}

int getsum(int x)
{
    int c=0;
    while (x>0)
    {
        c+=t[x];
        x-=lowbit(x);
    }
    return c;
}

void kp(int l,int r)
{
    if (l>r) return;
    int i=l,j=r;
    rec temp;
    rec key=b[(l+r)/2];
    while (i<=j)
    {
        while ((b[i].y<key.y) || (b[i].y==key.y) && (b[i].z<key.z) || (b[i].y==key.y) && (b[i].z==key.z) && (b[i].c<key.c)) i++;
        while ((b[j].y>key.y) || (b[j].y==key.y) && (b[j].z>key.z) || (b[j].y==key.y) && (b[j].z==key.z) && (b[j].c>key.c)) j--;
        if (i<=j)
        {
            temp=b[i]; b[i]=b[j]; b[j]=temp;
            i++; j--;
        }
    }
    kp(l,j);
    kp(i,r);
}

void cdq(int l,int r)
{
    if (l==r) return;
    int mid=(l+r)/2;
    cdq(l,mid);
    cdq(mid+1,r);
    for (int i=0;i<=r-l;i++)
    {
        b[i].y=a[i+l].y;
        b[i].z=a[i+l].z;
        if (i+l>mid) b[i].t=1;
                else b[i].t=0;
        b[i].c=i+l;        
    }           
    kp(0,r-l);
    for (int i=0;i<=r-l;i++)
    {
        if (!b[i].t) updata(b[i].z,1);
        else
        {
            a[b[i].c].ans+=getsum(b[i].z);
        }
    }
    for (int i=0;i<=r-l;i++)
    {
        if (!b[i].t) updata(b[i].z,-1);
    }
}

int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+n+1,cmp);
    cdq(1,n);   
    int k=0;
    for (int i=1;i<=n;i++)
    {
        if ((a[i].x==a[i+1].x) && (a[i].y==a[i+1].y) && (a[i].z==a[i+1].z)) k++;
        else
        {
            f[a[i].ans]+=k+1;
            k=0;
        }
    }
    for (int i=0;i<n;i++) printf("%d\n",f[i]);
}