BZOJ3262 陌上花开

标签：CDQ分治，树状数组

题目

题目传送门

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分治，将l->mid和mid+1->r区间分别按照第二维的关键字排序

第三维以树状数组为下标，维护每一朵花的出现次数

每一次处理[l,mid]对smid+1,r]的影响时，只需考虑第二维的影响

当第二维符合要求时，将它的影响加入树状数组中

每查找完[mid+1,r]区间的一个操作，就更新答案

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read()
{
    ll f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int maxn=1e5+6;
int tn,n,m,t[maxn<<1],ans[maxn];
struct node{int a,b,c,s,ans;}a[maxn],p[maxn];
inline bool cmp(node a,node b){
    if(a.a==b.a&&a.b==b.b)return a.c<b.c;
    return a.a==b.a?a.b<b.b:a.a<b.a;
}
inline bool operator < (node a,node b){return a.b==b.b?a.c<b.c:a.b<b.b;} 
inline int lowbit(int x){return x&(-x);}
void modify(int x,int num){for(int i=x;i<=m;i+=lowbit(i))t[i]+=num;}
inline int query(int x){int re=0;for(int i=x;i;i-=lowbit(i))re+=t[i];return re;}
void solve(int l,int r){
    if(l==r)return;int mid=(l+r)>>1;
    solve(l,mid);solve(mid+1,r);
    sort(p+l,p+mid+1);sort(p+mid+1,p+r+1);
    int i=l,j=mid+1;
    while(j<=r){
        while(i<=mid&&p[i].b<=p[j].b){modify(p[i].c,p[i].s);i++;}
        p[j].ans+=query(p[j].c);j++;
    }
    rep(j,l,i-1)modify(p[j].c,-p[j].s);
}
int main()
{
    tn=read(),m=read();
    rep(i,1,tn)a[i].a=read(),a[i].b=read(),a[i].c=read();
    sort(a+1,a+1+tn,cmp);int cnt=0;
    rep(i,1,tn){
        cnt++;
        if(a[i].a!=a[i+1].a||a[i].b!=a[i+1].b||a[i].c!=a[i+1].c){
            p[++n]=a[i];
            p[n].s=cnt;
            cnt=0;
        }
    }
    solve(1,n);
    rep(i,1,n)ans[p[i].ans+p[i].s-1]+=p[i].s;
    rep(i,0,tn-1)printf("%d\n",ans[i]);
    return 0;
}