bzoj3262 陌上花开——CDQ分治

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3262

第一道CDQ分治题!

看博客:https://www.cnblogs.com/Narh/p/9230515.html

CDQ分治意外地很好写啊,而且好厉害!膜拜CDQ!

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const maxn=1e5+5,maxm=2e5+5;
int n,m,f[maxm],ans[maxn],tot;
struct N{int a,b,c,sum,num;}s[maxn],t[maxn];
bool cmp(N x,N y){return (x.a<y.a) || (x.a==y.a && x.b<y.b) || (x.a==y.a && x.b==y.b && x.c<y.c);}
bool cmp2(N x,N y){return x.b<y.b;}
bool ck(N x,N y){return (x.a==y.a) && (x.b==y.b) && (x.c==y.c);}
void add(int x,int v){for(;x<=m;x+=(x&-x))f[x]+=v;}// <=m 而非 <=n !
int query(int x){int ret=0; for(;x;x-=(x&-x))ret+=f[x]; return ret;}
void cdq(int l,int r)
{
    if(l==r)return;
    int mid=((l+r)>>1);
    cdq(l,mid); cdq(mid+1,r);
    sort(t+l,t+mid+1,cmp2); sort(t+mid+1,t+r+1,cmp2);
    int p=l;
    for(int i=mid+1;i<=r;i++)
    {
        while(p<=mid && t[p].b<=t[i].b)add(t[p].c,t[p].num),p++;// <=
        t[i].sum+=query(t[i].c);
    }
    for(int i=l;i<p;i++)add(t[i].c,-t[i].num);//撤销 
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d%d%d",&s[i].a,&s[i].b,&s[i].c);
    sort(s+1,s+n+1,cmp);
    for(int i=1,cnt;i<=n;i++)
    {
        cnt=1;
        while(ck(s[i],s[i+1]))cnt++,i++;
        t[++tot]=s[i]; t[tot].num=cnt;
    }
    cdq(1,tot);
    for(int i=1;i<=tot;i++)ans[t[i].sum+t[i].num-1]+=t[i].num;
    for(int i=0;i<n;i++)printf("%d\n",ans[i]);
    return 0;
}

 

posted @ 2018-06-26 20:19  Zinn  阅读(125)  评论(0编辑  收藏  举报