bzoj2754: [SCOI2012]喵星球上的点名 （后缀数组+树状数组+莫队）

哇哇哇哇！我终于…改对了！！！
太不容易了！！！！！

这题啊…..据说….ac自动机模板或者后缀数组暴力就可以了…然后..也确实可以…
但是…亲爱的洛谷！他..加了一个测试数据！然后就…过不去了【再见】
所以…就要…改进…十分凶残（好久没写这么长的了qaq）

这题第一问是求每个点名串是多少个姓名串的字串
所以呢..我们可以用后缀数组求出height数组（这里要把姓名串和点名串连起来一起求）
然后..找出每个点名串可以延伸到哪里（这里用rmq+二分求……）
那么我们可以找出一段区间能被这个点名串点到，所以能点到几个人就是这段区间里有多少个人。这个可以用树状数组求种类数来求（不要脸地推此文章：树状数组求不同种类数）
我这里用的莫队…因为….没想到树状数组的做法【再见】

第二问是求每个人会被多少个点到
这次呢..对于每个点名串能点到的区间 $[l,r]$ ，在 $l$ 处加1，$r+1$ 处减1，那么 $i$ 的贡献就是$query(i)-query(last[num[i]])$
树状数组求前缀和即可

悄咪咪的说…好长

#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 440000
#define M 10010
int n,m,tot=0,num[N],len1[N],cnt[N],rank[N<<1],rank1[N<<1],tmp[N],sa[N],height[N],f[N][30],lg[N],flag[N],tree[N],last[N];
struct node{int x,num,f;}a[N];//数值，第几个串，模式串还是闻讯串 
struct node1{int l,r,q,num;}b[N];
bool cmp(node1 x,node1 y){return x.q<y.q || (x.q==y.q && x.r<y.r);}
bool cmp1(node1 x,node1 y){return x.r<y.r;}
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while('0'<=ch && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
void rmq(){                 //rmq求区间最小值
    for(int i=1;i<=tot;i++) f[i][0]=height[i];
    lg[1]=0;for(int i=2;i<=tot;i++) lg[i]=lg[i>>1]+1;
    for(int j=1;j<=lg[tot];j++)
        for(int i=1;i<=tot+1-(1<<j);i++)
            f[i][j]=min(f[i][j-1],f[i+(1<<j-1)][j-1]);
}
int find(int l,int r){      
    int k=lg[r-l+1];
    return min(f[l][k],f[r+1-(1<<k)][k]); 
}
int findl(int r,int len){       //二分找左端点
    int l=1,ans=r+1,rr=r;
    while(l<=r){
        int mid=l+r>>1;
        if(find(mid,rr)>=len) r=mid-1,ans=mid;
        else l=mid+1;
    }return ans-1;
}
int findr(int l,int len){       //二分找右端点
    int r=tot,ans=l;l++;int ll=l;
    while(l<=r){
        int mid=l+r>>1;
        if(find(ll,mid)>=len) l=mid+1,ans=mid;
        else r=mid-1;
    }return ans;
}
void add(int x,int y){
    for(;x<=tot;x+=x&-x) tree[x]+=y;
}
int query(int x){
    if(x<=0) return 0; 
    int y=0;for(;x;x-=x&-x) y+=tree[x];
    return y;
}
int main(){
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    n=read();m=read();
    for(int i=1;i<=n;i++){
        int x=read();for(int j=1;j<=x;j++) a[++tot].x=read(),a[tot].num=i,a[tot].f=1;
        a[++tot].x=10001;
        x=read();for(int j=1;j<=x;j++) a[++tot].x=read(),a[tot].num=i,a[tot].f=1;
        a[++tot].x=10001;
    }for(int i=1;i<=m;i++){
        len1[i]=read();num[i]=tot+1;
        for(int j=1;j<=len1[i];j++) a[++tot].x=read(),a[tot].num=i,a[tot].f=2;
        a[++tot].x=10001;
    }

    for(int i=1;i<=tot;i++) cnt[a[i].x]=1;         //后缀数组求sa、rank
    for(int i=1;i<=10001;i++) cnt[i]+=cnt[i-1];
    for(int i=1;i<=tot;i++) rank[i]=cnt[a[i].x];
    int k=0;
    for(int len=1;k!=tot;len<<=1){
        memset(cnt,0,sizeof(cnt));
        for(int i=1;i<=tot;i++) cnt[rank[i+len]]++;
        for(int i=1;i<=tot;i++) cnt[i]+=cnt[i-1];
        for(int i=tot;i>0;i--) tmp[cnt[rank[i+len]]--]=i;

        memset(cnt,0,sizeof(cnt));
        for(int i=1;i<=tot;i++) cnt[rank[tmp[i]]]++;
        for(int i=1;i<=tot;i++) cnt[i]+=cnt[i-1];
        for(int i=tot;i>0;i--) sa[cnt[rank[tmp[i]]]--]=tmp[i];

        memcpy(rank1,rank,sizeof(rank));
        rank[sa[1]]=k=1;
        for(int i=2;i<=tot;i++){
            if(rank1[sa[i]]!=rank1[sa[i-1]] || rank1[sa[i]+len]!=rank1[sa[i-1]+len]) k++;
            rank[sa[i]]=k;
        }
    }k=0;
    for(int i=1;i<=tot;i++){       //求height
        if(rank[i]==1) k=0;
        else{
            if(--k<0) k=0;
            while(a[i+k].x==a[sa[rank[i]-1]+k].x) k++;
        }height[rank[i]]=k;
    }rmq();

    int qq=sqrt(tot);                 //莫队
    for(int i=1;i<=m;i++){
        b[i].num=i;
        b[i].l=findl(rank[num[i]],len1[i]);
        b[i].r=findr(rank[num[i]],len1[i]);
        b[i].q=(b[i].l-1)/qq+1;
    }sort(b+1,b+m+1,cmp);
    memset(cnt,0,sizeof(cnt));
    memset(flag,false,sizeof(flag));
    int l=1,r=0,ans=0;
    for(int i=1;i<=m;i++){
        while(l<b[i].l){
            int x=sa[l];l++;
            if(a[x].f==1){
                flag[a[x].num]--;
                if(!flag[a[x].num]) ans--;
            }
        }
        while(l>b[i].l){
            l--;int x=sa[l];
            if(a[x].f==1){
                if(!flag[a[x].num]) ans++;
                flag[a[x].num]++;
            }
        }
        while(r>b[i].r){
            int x=sa[r];r--;
            if(a[x].f==1){
                flag[a[x].num]--;
                if(!flag[a[x].num]) ans--;
            }
        }
        while(r<b[i].r){
            r++;int x=sa[r];
            if(a[x].f==1){
                if(!flag[a[x].num]) ans++;
                flag[a[x].num]++;
            }
        }cnt[b[i].num]=ans;
    }for(int i=1;i<=m;i++) printf("%d\n",cnt[i]);   

    memset(cnt,0,sizeof(cnt));
    for(int i=1;i<=m;i++) add(b[i].l,1);
    sort(b+1,b+m+1,cmp1);
    int now=1;
    for(int i=1;i<=tot;i++){
        while(now<=m && b[now].r<i){
            add(b[now].l,-1);
            now++;
        }if(a[sa[i]].f==1) cnt[a[sa[i]].num]+=query(i)-query(last[a[sa[i]].num]),last[a[sa[i]].num]=i;
    } 
    for(int i=1;i<n;i++) printf("%d ",cnt[i]);printf("%d",cnt[n]);
    return 0;
}