[SCOI2012]喵星球上的点名

2754: [SCOI2012]喵星球上的点名

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 1953  Solved: 862
[Submit][Status][Discuss]

Description

a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。   假设课堂上有N个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。 然而，由于喵星人的字码过于古怪，以至于不能用ASCII码来表示。为了方便描述，a180285决定用数串来表示喵星人的名字。

现在你能帮助a180285统计每次点名的时候有多少喵星人答到，以及M次点名结束后每个喵星人答到多少次吗？  

Input

 

现在定义喵星球上的字符串给定方法：

先给出一个正整数L，表示字符串的长度，接下来L个整数表示字符串的每个字符。

输入的第一行是两个整数N和M。

接下来有N行，每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的

字符串。

接下来有M行，每行包含一个喵星球上的字符串，表示老师点名的串。

Output

 

对于每个老师点名的串输出有多少个喵星人应该答到。

然后在最后一行输出每个喵星人被点到多少次。

Sample Input

2 3
6 8 25 0 24 14 8 6 18 0 10 20 24 0
7 14 17 8 7 0 17 0 5 8 25 0 24 0
4 8 25 0 24
4 7 0 17 0
4 17 0 8 25

Sample Output

2
1
0
1 2
【提示】
事实上样例给出的数据如果翻译成地球上的语言可以这样来看
2 3
izayoi sakuya
orihara izaya
izay
hara
raiz

HINT

【数据范围】 

 对于30%的数据，保证： 

1<=N,M<=1000，喵星人的名字总长不超过4000，点名串的总长不超过2000。

对于100%的数据，保证：

1<=N<=20000，1<=M<=50000，喵星人的名字总长和点名串的总长分别不超过100000，保证喵星人的字符串中作为字符存在的数不超过10000。

 

#include<cstdio>
#include<cstring>
#define p 29
#define debug(x) printf("%I64u ",x);
using namespace std;
typedef unsigned long long UL;
const int N=1005;
UL chk,Q[N<<1],ts[N<<2],hs[N][N<<2],a[N<<1],base[N];
int n,m,xlen[2][N<<1],len[N<<1];
int Ans[N<<1],ans[N<<1];
inline int read(){
    int x=0,f=1;char ch=getchar();
    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;
}
UL query(int x,int y,int id){
    return hs[id][y]-hs[id][x-1]*base[y-x+1];
}
int main(){
    freopen("name.in","r",stdin);
    freopen("name.out","w",stdout);
    n=read();m=read();
    for(int i=1,x1;i<=n;i++){
        x1=read();xlen[0][i]=x1;hs[i][0]=0;
        for(int j=1;j<=x1;j++) a[j]=read();
        for(int j=0;j<=x1;j++) ts[j]=0;
        for(int j=1;j<=x1;j++) ts[j]=ts[j-1]*p+a[j];
        for(int j=1;j<=x1;j++) hs[i][j]=ts[j];
        x1=read();xlen[1][i]=x1;hs[i][xlen[0][i]+1]=0;
        for(int j=1;j<=x1;j++) a[j]=read();
        for(int j=0;j<=x1;j++) ts[j]=0;
        for(int j=1;j<=x1;j++) ts[j]=ts[j-1]*p+a[j];
        for(int j=1;j<=x1;j++) hs[i][j+xlen[0][i]+1]=ts[j];
    }
    for(int i=1,x1;i<=m;i++){
        x1=read();len[i]=x1;
        for(int j=1;j<=x1;j++) a[j]=read();
        for(int j=1;j<=x1;j++) Q[i]=Q[i]*p+a[j];
    }
    base[0]=1;
    for(int i=1;i<=100;i++) base[i]=base[i-1]*p;
    for(int i=1;i<=m;i++){//点名的串
        for(int k=1;k<=n;k++){//姓和名两个串
            int flag=0;
            for(int j=1;j<=xlen[0][k]-len[i]+1;j++){
                chk=query(j,j+len[i]-1,k);
                if(chk==Q[i]){
                    Ans[i]++;ans[k]++;flag=1;break;
                }
            }
            if(flag) continue; 
            for(int j=1;j<=xlen[1][k]-len[i]+1;j++){
                chk=query(j+xlen[0][k]+1,j+xlen[0][k]+len[i],k);
                if(chk==Q[i]){
                    Ans[i]++;ans[k]++;break;
                }
            }
        }
    }
    for(int i=1;i<=m;i++) printf("%d
",Ans[i]);
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
    return 0;
}

考试的时候，直觉告诉我:可能是AC自动机（“子串”一词）

然，离线把询问串见trie，然后对每个名字串进行匹配。（反着搞）

O(20000*100000)

写出来我就后悔了。（样例还没过去）

算了，30分，暴力hash。

试后，网上正解fail数，反着的AC自动机（我去~真跟AC自动机有关~后悔当时没多想想）

看了半天，fail树弃疗了（感觉他的性质好神奇）

民间还是有向我一样的蒟蒻的，把所有名字串连起来，用后缀数组询问子串。

又有M组询问（好烦人~MO DEI KEI），划分一下，离线处理，至此bingo

#include<cstdio>
#include<cstring>
using namespace std;
const int N=5e5+7,M=5e4+7;
int n,m,len,maxx,sa[N],tsa[N],rank[N],trank[N],h[N],c[N];
int fengefu,crs[N],s[N],ans[N];int vis[M];
struct no{int n,s;}Q[M];char st[N];
void init(){
    fengefu=10100;
    scanf("%d%d",&n,&m);
    for(int i=1,k;i<=n;i++){
        scanf("%d",&k);
        for(int j=1;j<=k;j++){
            crs[++len]=i;
            scanf("%d",&s[len]);s[len]++;
        }
        s[++len]=++fengefu;
        scanf("%d",&k);
        for(int j=1;j<=k;j++){
            crs[++len]=i;
            scanf("%d",&s[len]);s[len]++;
        }
        s[++len]=++fengefu;
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&Q[i].n);Q[i].s=len;
        for(int j=1;j<=Q[i].n;j++){
            scanf("%d",&s[++len]);s[len]++;
        }
        s[++len]=++fengefu;
    }
}
void DA(int maxx=256){
    int p;
    memset(c,0,sizeof c);
    for(int i=1;i<=len;i++) c[rank[i]=s[i]]++;
    for(int i=2;i<=maxx;i++) c[i]+=c[i-1];
    for(int i=len;i;i--) sa[c[rank[i]]--]=i;
    trank[sa[1]]=p=1;
    for(int i=2;i<=len;i++){
        if(rank[sa[i]]!=rank[sa[i-1]]) p++;
        trank[sa[i]]=p;
    }
    for(int i=1;i<=len;i++) rank[i]=trank[i];
    for(int k=1;p<len;k<<=1,maxx=p){
        p=0;
        for(int i=len-k+1;i<=len;i++) tsa[++p]=i;
        for(int i=1;i<=len;i++) if(sa[i]>k) tsa[++p]=sa[i]-k;
        memset(c,0,sizeof c);
        for(int i=1;i<=len;i++) trank[i]=rank[tsa[i]];
        for(int i=1;i<=len;i++) c[trank[i]]++;
        for(int i=2;i<=maxx;i++) c[i]+=c[i-1];
        for(int i=len;i;i--) sa[c[trank[i]]--]=tsa[i];
        trank[sa[1]]=p=1;
        for(int i=2;i<=len;i++){
            if(rank[sa[i]]!=rank[sa[i-1]]||rank[sa[i]+k]!=rank[sa[i-1]+k]) p++;
            trank[sa[i]]=p;
        }
        for(int i=1;i<=len;i++) rank[i]=trank[i];
    }
    for(int i=1,k=0;i<=len;i++){
        int j=sa[rank[i]-1];
        while(s[i+k]==s[j+k]) k++;
        h[rank[i]]=k;if(k>0) k--;
    }
}
void work(){
    for(int i=1,l,r,res;i<=m;i++){
        l=r=rank[Q[i].s+1];
        while(h[l]>=Q[i].n) l--;
        while(h[r]>=Q[i].n) r++;
        r--;
        res=0;
        for(int j=l;j<=r;j++){
            if(crs[sa[j]]){
                if(vis[crs[sa[j]]]!=i){
                    vis[crs[sa[j]]]=i;
                    res++;
                    ans[crs[sa[j]]]++;
                }
            }
        }
        printf("%d
",res);
    }
    for(int i=1;i<=n;i++) printf("%d ",ans[i]);
}
int main(){
    freopen("name.in","r",stdin);
    freopen("name.out","w",stdout);
    init();
    DA(200000);
    work();
    return 0;
}

【暴力AC自动机思路】：

考虑把点名串建tire树，打上末尾标记，构建失败指针，。

然后姓和名用-1连成一个串S（防止姓的后面和名的前面组成串在tire中）

若匹配到末尾标记，记录点名和被点名的贡献。

跑m次匹配。输出ans

【注意】：

　　节点数较多，要用map

　　使用vector替代数组

//暴力map+ACmach 可过 
//正解：后缀数组+划分 
#include<cstdio>
#include<map>
#include<vector>
#define Sz 10000
using namespace std;
const int N=1e5+5;
typedef map<int,int> MAP;
MAP tr[N];
int n,m,cnt=1,fail[N],q[N];bool vis[N],mark[N];
int ans1[N],ans2[N];
vector<int> tag[N],a[N],V,M;
inline int read(){
    int x=0,f=1;char ch=getchar();
    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;
}
inline void insert(int id){
    int now=1,l=read();
    for(int i=0,z;i<l;i++){
        z=read();
        if(!tr[now][z]) tr[now][z]=++cnt;
        now=tr[now][z];
    }
    tag[now].push_back(id);
}
inline void acmach(){
    for(int i=-1;i<Sz;i++) tr[0][i]=1;
    int h=0,t=1,now,z,p;q[t]=1;fail[1]=0;
    while(h!=t){
        now=q[++h];
        for(MAP::iterator i=tr[now].begin();i!=tr[now].end();i++){
            z=i->first;
            if(!i->second) continue;
            p=fail[now];
            while(!tr[p][z]) p=fail[p];
            fail[i->second]=tr[p][z];
            q[++t]=i->second;
        }
    }
}
inline void get(int id,int now){
    for(int i=now;i;i=fail[i]){
        if(!vis[i]){
            vis[i]=1;V.push_back(i);
            for(int j=0;j<tag[i].size();j++){
                if(!mark[tag[i][j]]){
                    mark[tag[i][j]]=1;M.push_back(tag[i][j]);
                    ans1[tag[i][j]]++;
                    ans2[id]++;
                }
            }
        }
        else break;
    }
}
inline void solve(int id){
    int z,now=1,l=a[id].size();
    for(int i=0;i<l;i++){
        z=a[id][i];
        while(!tr[now][z]) now=fail[now];
        now=tr[now][z];
        get(id,now);
    }
    for(int i=0;i<V.size();i++) vis[V[i]]=0;
    for(int i=0;i<M.size();i++) mark[M[i]]=0;
    V.clear();M.clear();
}
int main(){
    n=read();m=read();
    for(int i=1,x,l;i<=n;i++){
        l=read();
        while(l--) x=read(),a[i].push_back(x);
        a[i].push_back(-1); 
        l=read();
        while(l--) x=read(),a[i].push_back(x);
    }
    for(int i=1;i<=m;i++) insert(i);
    acmach();
    for(int i=1;i<=n;i++) solve(i);
    for(int i=1;i<=m;i++) printf("%d
",ans1[i]);
    for(int i=1;i<=n;i++) printf("%d ",ans2[i]);
    return 0;
}