[SCOI2012]喵星球上的点名
2754: [SCOI2012]喵星球上的点名
Time Limit: 20 Sec Memory Limit: 128 MBSubmit: 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
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\n",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\n",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\n",ans1[i]); for(int i=1;i<=n;i++) printf("%d ",ans2[i]); return 0; }
