题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2754
大意:给n个串,每个串分为2部分(姓和名)。串由(<=10000)的数字组成。再给m个串,求这m个串分别是n个串子串的个数,以及n个串分别包含m个串的个数。
感觉自己最近真的是极度拖延症。并且代码能力也是boom!boom!boom!每次看到难打的题都是只想想就扔一边了。(这是退役的前兆吗。。。)
回归正题,直接将所有串连起来(要注意分隔符要不同)。然后后缀数组+暴力判断。(听说很难卡的)。
(还有种AC自动机的方法,好像也是很暴力)
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,num,Mark[500050],s[500050],pos[500005],Len[500050];
int rank[500050],sa[500050],w[500040],hi[500050];
int vis[500050],ans[500050];
inline bool In_equal(int *x,int i,int j,int k)
{
int tx = (i+k<num) ? x[i+k] : -1;
int ty = (j+k<num) ? x[j+k] : -1;
return (x[i]==x[j]&&tx==ty);
}
inline void Make_Sa()
{
int *x=rank,*y=hi,r=10013,yn;
for (int i=0;i<r;++i) w[i]=0;
for (int i=0;i<num;++i) ++w[s[i]];
for (int i=1;i<r;++i) w[i]+=w[i-1];
for (int i=num-1;i>=0;--i) sa[--w[s[i]]]=i;
x[sa[0]]=0; r=1;
for (int i=1;i<num;++i)
x[sa[i]]= (s[sa[i]]==s[sa[i-1]]) ? r-1 : r++;
for (int k=1;r<num;k<<=1) {
yn=0;
for (int i=num-k;i<num;++i) y[yn++]=i;
for (int i=0;i<num;++i)
if (sa[i]>=k) y[yn++]=sa[i]-k;
for (int i=0;i<r;++i) w[i]=0;
for (int i=0;i<num;++i) ++w[x[y[i]]];
for (int i=1;i<r;++i) w[i]+=w[i-1];
for (int i=num-1;i>=0;--i) sa[--w[x[y[i]]]]=y[i];
swap(x,y); r=1; x[sa[0]]=0;
for (int i=1;i<num;++i)
x[sa[i]]= In_equal(y,sa[i],sa[i-1],k) ? r-1 : r++;
}
if (x!=rank) copy(x,x+num,rank);
hi[0]=0;
for (int i=0,k=0,j;i<num;++i) {
if (k) --k;
if (!rank[i]) continue;
j=sa[rank[i]-1];
while (i+k<num&&j+k<num&&s[i+k]==s[j+k]) ++k;
hi[rank[i]]=k;
}
}
inline void Work()
{
int l,r,tmp;
for (int i=1;i<=m;++i) {
l=r=rank[pos[i]];
while (hi[l]>=Len[i]) --l;
while (hi[r]>=Len[i]) ++r;
--r; tmp=0;
for (int j=l;j<=r;++j)
if (Mark[sa[j]]&&vis[Mark[sa[j]]]!=i) {
vis[Mark[sa[j]]]=i; ++tmp;
++ans[Mark[sa[j]]];
}
printf("%d\n",tmp);
}
for (int i=1;i<=n;++i) {
printf("%d",ans[i]);
if (i<n) putchar(' ');
else printf("\n");
}
}
int main()
{
scanf("%d%d",&n,&m);
int k,x;
for (int i=1;i<=n;++i) {
scanf("%d",&k);
while (k--) {
scanf("%d",&x);
Mark[num]=i;
s[num++]=x;
}
s[num++]=10001;
scanf("%d",&k);
while (k--) {
scanf("%d",&x);
Mark[num]=i;
s[num++]=x;
}
s[num++]=10001;
}
for (int i=1;i<=m;++i) {
scanf("%d",&k);
Len[i]=k;
pos[i]=num;
while (k--) {
scanf("%d",&x);
s[num++]=x;
}
s[num++]=10011;
}
Make_Sa();
Work();
}