【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 110000
inline 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;
}