[BZOJ3262]陌上花开（cdq分治+bit）

题目描述

传送门

题解

三维偏序问题？
首先按照x第一关键字，y第二关键字，z第三关键字排序，然后问题就转化成了求某一个(x,y,z)前面存在多少个(x’,y’,z)满足x>=x’,y>=y’,z>=z’
由于x一定是有序的，那么根据x进行cdq分治
每一次将(l,mid)和(mid+1,r)分别根据y排序（x不一定有序），然后两个指针扫一下，查询之前把z小于等于它的都插入到bit中
这样做完了之后还有一个问题是有一些(x,y,z)是完全相同的，这个时候只有排在最后一个的统计的是全部的，所以从后往前max一遍
最后统计答案就可以了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 200005

int n,k,acnt,bcnt,pa,pb,tot;
struct hp{int x,y,z,id,ans;}f[N],a[N],b[N];
int C[N],cnt[N],ch[N];

int cmp1(hp a,hp b)
{
    return a.x<b.x||(a.x==b.x&&a.y<b.y)||(a.x==b.x&&a.y==b.y&&a.z<b.z);
}
int cmp2(hp a,hp b)
{
    return a.y<b.y;
}
void add(int loc,int val)
{
    if (!loc) return;
    for (int i=loc;i<=k;i+=i&(-i))
        C[i]+=val;
}
int query(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans+=C[i];
    return ans;
}
void cdq(int l,int r)
{
    if (l>=r) return;
    int mid=(l+r)>>1;
    cdq(l,mid);

    acnt=0;
    for (int i=l;i<=mid;++i) a[++acnt]=f[i];
    sort(a+1,a+acnt+1,cmp2);
    bcnt=0;
    for (int i=mid+1;i<=r;++i) b[++bcnt]=f[i];
    sort(b+1,b+bcnt+1,cmp2);
    pa=pb=1;tot=0;
    while (pb<=bcnt)
    {
        while (pa<=acnt&&a[pa].y<=b[pb].y)
        {
            add(a[pa].z,1);
            ch[++tot]=a[pa].z;
            ++pa;
        }
        f[b[pb].id].ans+=query(b[pb].z);
        ++pb;
    }
    for (int i=1;i<=tot;++i)
        add(ch[i],-1);

    cdq(mid+1,r);
}
int main()
{
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;++i)
        scanf("%d%d%d",&f[i].x,&f[i].y,&f[i].z);
    sort(f+1,f+n+1,cmp1);
    for (int i=1;i<=n;++i) f[i].id=i;
    cdq(1,n);
    for (int i=n-1;i>=1;--i)
        if (f[i].x==f[i+1].x&&f[i].y==f[i+1].y&&f[i].z==f[i+1].z)
            f[i].ans=max(f[i].ans,f[i+1].ans);
    for (int i=1;i<=n;++i)
        ++cnt[f[i].ans];
    for (int i=0;i<n;++i)
        printf("%d\n",cnt[i]);
}