[cdq分治][树状数组] Bzoj P3262 陌上花开

Description

有n朵花,每朵花有三个属性:花形(s)、颜色(c)、气味(m),用三个整数表示。

现在要对每朵花评级,一朵花的级别是它拥有的美丽能超过的花的数量。

定义一朵花A比另一朵花B要美丽,当且仅Sa>=Sb,Ca>=Cb,Ma>=Mb。

显然,两朵花可能有同样的属性。需要统计出评出每个等级的花的数量。

 

Input

第一行为N,K (1 <= N <= 100,000, 1 <= K <= 200,000 ), 分别表示花的数量和最大属性值。

以下N行，每行三个整数si, ci, mi (1 <= si, ci, mi <= K)，表示第i朵花的属性

Output

包含N行，分别表示评级为0...N-1的每级花的数量。

Sample Input

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

Sample Output

3
1
3
0
1
0
1
0
0
1

题解

• 这显然就是个三维偏序问题
• 就先排序，cdq套树状数组就好了

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=200001;
struct node{ int x,y,z,id; }a[N];
int c[N<<2],k,n,b[N],bj[N],f[N];
int lowbit(int x){ return x&(-x); }
void add(int x,int v){ while(x<=k) c[x]+=v,x+=lowbit(x); }
int sum(int x)
{
    int ans=0;
    while (x) ans+=c[x],x-=lowbit(x);
    return ans;
}
bool cmp1(const node &a,const node &b)
{
    if (a.x!=b.x) return a.x<b.x;
    if (a.y!=b.y) return a.y<b.y;
    return a.z<b.z;
}
bool cmp2(const node &a,const node &b)
{
    if (a.y!=b.y) return a.y<b.y;
    if (a.z!=b.z) return a.z<b.z;
    return a.x<b.x;
}
void cdq(int l,int r)
{
    if (l==r) return;
    int mid=(l+r)>>1,flag;
    cdq(l,mid),cdq(mid+1,r);
    sort(a+l,a+r+1,cmp2);
    for (int i=l;i<=r;i++) (a[i].x<=mid)?add(a[i].z,1),flag=i:b[a[i].id]+=sum(a[i].z);
    for (int i=l;i<=r;i++) if (a[i].x<=mid) add(a[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),a[i].id=i;
    sort(a+1,a+1+n,cmp1);
    for (int i=1;i<=n;)
    {
        int j=i+1;
        while (j<=n&&a[j].x==a[i].x&&a[j].y==a[i].y&&a[j].z==a[i].z) j++;
        while (i<j) bj[a[i].id]=a[j-1].id,i++;
    }
    for(int i=1;i<=n;i++) a[i].x=i;
    cdq(1,n);
    for(int i=1;i<=n;i++) f[b[bj[a[i].id]]]++;
    for(int i=0;i<n;i++) printf("%d
",f[i]);
}