【BZOJ3262】陌上花开（CDQ分治）

题解

原来放过这道题目，题面在这里
树套树的做法也请点上面

这回用CDQ分治做的
其实也很简单，
对于第一维排序之后
显然只有前面的对后面的才会产生贡献

那么，使用CDQ分治
先分，每次递归子问题
合并的时候每次考虑前面的对于后面的贡献
最后统计一下答案

如果在清空树状数组的时候用了memset会TLE

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<cmath>#include<algorithm>#include<set>#include<map>#include<queue>#include<vector>using namespace std;#define MAX 110000inline int read(){    int x=0,t=1;char ch=getchar();    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();    if(ch=='-')t=-1,ch=getchar();    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();    return x*t;}int c[MAX<<1],n,K;int ans[MAX];struct TNode{int a,b,c,w,ans;}t[MAX];bool operator<(TNode x,TNode y){    if(x.a!=y.a)return x.a<y.a;    if(x.b!=y.b)return x.b<y.b;    return x.c<y.c;}bool operator==(TNode a,TNode b){    return a.a==b.a&&a.b==b.b&&a.c==b.c;}bool cmp(TNode x,TNode y){    if(x.b!=y.b)return x.b<y.b;    return x.c<y.c;}inline int lowbit(int x){return x&(-x);}inline void Add(int x,int w){while(x<=K)c[x]+=w,x+=lowbit(x);}inline int Getsum(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;    CDQ(l,mid);CDQ(mid+1,r);    sort(&t[l],&t[mid+1],cmp);    sort(&t[mid+1],&t[r+1],cmp);    int i=l;    for(int j=mid+1;j<=r;++j)    {        while(i<=mid&&t[j].b>=t[i].b)        {            Add(t[i].c,t[i].w);            ++i;        }        t[j].ans+=Getsum(t[j].c);    }    for(int j=l;j<i;++j)Add(t[j].c,-t[j].w);    //memset(c,0,sizeof(c));}int main(){     n=read();K=read();    for(int i=1;i<=n;++i)        t[i].a=read(),t[i].b=read(),t[i].c=read();    sort(&t[1],&t[n+1]);    int tot=0;    for(int i=1;i<=n;++i)    {        if(t[i]==t[i-1])++t[tot].w;        else t[++tot]=t[i],t[tot].w++;    }    int N=n;    n=tot;    CDQ(1,n);           for(int i=1;i<=n;++i)ans[t[i].ans+t[i].w-1]+=t[i].w;    for(int i=0;i<N;++i)        printf("%d\n",ans[i]);    return 0;}