bzoj 3262: 陌上花開 (cdq分治)

3262: 陌上花開

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 1700  Solved: 742
[Submit][Status][Discuss]

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

Source

樹套樹 CDQ分治

複習cdq

排序搞一維，cdq降一維，樹狀數組處理一維

代碼：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<climits>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define N 400020
#define lowbit(x) x&-x
using namespace std;
inline int read()
{
    int x=0,f=1;char ch;
    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;
}
int n,m,tot;
struct node{int a,b,c,d,num;}p[N],tmp[N];
bool operator < (node a,node b)
{
    if(a.a!=b.a)return a.a<b.a;
    if(a.b!=b.b)return a.b<b.b;
    return a.c<b.c;
}
bool cmp(node a,node b)
{
    if(a.b!=b.b)return a.b<b.b;
    if(a.d!=b.d)return a.d<b.d;
    return a.c<b.c;
}
int C[N],ans[N];
void add(int x,int val)
{
    while(x<=m)
    {
	C[x]+=val;
	x+=lowbit(x);
    }
}
int ask(int x)
{
    int ret=0;
    while(x)
    {
	ret+=C[x];
	x-=lowbit(x);
    }return ret;
}
void cdq(int l,int r)
{
    if(l==r)return ;int mid=(l+r)>>1;
    for(int i=l;i<=r;i++)
		if(p[i].d<=mid)add(p[i].c,p[i].num);
		else ans[p[i].d]+=ask(p[i].c);
    int L=l,R=mid+1;
    for(int i=l;i<=r;i++)
	if(p[i].d<=mid)
	{
	    add(p[i].c,-p[i].num);
	    tmp[L++]=p[i];
	}
	else tmp[R++]=p[i];
	for(int i=l;i<=r;i++)
		p[i]=tmp[i];
    cdq(l,mid);cdq(mid+1,r);
}
int ret[N];
int main()
{
//	freopen("in.txt","r",stdin);
//	freopen("my.txt","w",stdout);
    n=read(),m=read();
    for(int i=1;i<=n;i++)
	p[i].a=read(),p[i].b=read(),p[i].c=read();
    sort(p+1,p+1+n);p[++tot]=p[1];p[tot].d=1,p[tot].num=1;
    for(int i=2;i<=n;i++)
    {
		if(p[i].a==p[i-1].a&&p[i].b==p[i-1].b&&p[i].c==p[i-1].c)
			p[tot].num++;
		else p[++tot]=p[i],p[tot].num=1,p[tot].d=tot;
    }sort(p+1,p+1+tot,cmp);for(int i=1;i<=tot;i++)ans[p[i].d]=p[i].num-1;
    cdq(1,tot);
//    for(int i=1;i<=tot;i++)
 //   	printf("%d %d %d : %d\n",p[i].a,p[i].b,p[i].c,ans[p[i].d]);
	for(int i=1;i<=tot;i++)
		ret[ans[i]]+=p[i].num;
	for(int i=0;i<n;i++)
		printf("%d\n",ret[i]);
}