【bzoj3262】【陌上花开】【树状数组套平衡树】

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

HINT

1 <= N <= 100,000, 1 <= K <= 200,000

题解：

           按s大小排序,按c大小建树状数组,树状数组里套以m为关键字的平衡树.

           因为相等的情况要算进去,所以对于相同的s要全加进去再查询.

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#define N 100010
#define M 200010
using namespace std;
int ls[N*30],rs[N*30],root[M],sum[N],n,ans,m,sz;
struct use{int x,y,z;}a[N];
struct treap{int rd,w,s,v;}tr[M*30];
bool cmp(use a,use b){return a.x<b.x;}
void update(int k){
  tr[k].s=tr[ls[k]].s+tr[rs[k]].s+tr[k].w;
}
void lturn(int &k){
  int t=rs[k];rs[k]=ls[rs[k]];ls[t]=k;
  update(k);update(t);k=t;
}
void rturn(int &k){
  int t=ls[k];ls[k]=rs[ls[k]];rs[t]=k;
  update(k);update(t);k=t;
}
void insert(int &k,int v){
  if (!k){k=++sz;tr[k].s=tr[k].w=1;tr[k].v=v;tr[k].rd=rand();return;}
  if (tr[k].v==v) tr[k].w++;
  else if (v<tr[k].v){insert(ls[k],v);if (tr[ls[k]].rd<tr[k].rd) rturn(k);}
  else{insert(rs[k],v);if (tr[rs[k]].rd<tr[k].rd) lturn(k);}
  update(k);
}
void ask(int k,int v){
  if (!k) return;
  if (tr[k].v==v) ans+=tr[ls[k]].s+tr[k].w;
  else if (v<tr[k].v) ask(ls[k],v);
  else{ans+=tr[ls[k]].s+tr[k].w;ask(rs[k],v);}
}
void que(int x,int y){for (int i=x;i;i-=i&(-i)) ask(root[i],y);}
void add(int x,int y){for (int i=x;i<=m;i+=i&(-i)) insert(root[i],y);}
int main(){
   //freopen("a.in","r",stdin);
   scanf("%d%d",&n,&m);
   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);int t=1;
   while (t<=n){
     int j;
     for (j=t;a[j].x==a[t].x;j++) add(a[j].y,a[j].z);j--;
     for (int i=t;i<=j;i++){
       ans=-1;que(a[i].y,a[i].z);sum[ans]++;
       //cout<<a[i].x<<' '<<a[i].y<<' '<<ans<<endl;
     }
     t=j+1;
   }
   for (int i=0;i<=n-1;i++) printf("%d\n",sum[i]);
}