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

来源：互联网发布：mysql 增加unique 编辑：程序博客网时间：2024/05/29 16:22

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 200010using 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]);}

0 0