bzoj2754 [SCOI2012]喵星球上的点名

传送门

脑残错误调了将近一小时……这人没救了

……

其实跟bzoj3881差不多，建出AC自动机之后跑树链的并即可。因为只要出现姓或名就算一次，因此要把两次匹配得到的所有点一起做树链的并。

/**************************************************************
Problem: 2754
User: hzoier
Language: C++
Result: Accepted
Time:1048 ms
Memory:24460 kb
****************************************************************/
#include<cstdio>
#include<cstring>
#include<stdint.h>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
const int maxn=200010;
int insert(const int*,int);
void getfail();
void bfs();
int LCA(int,int);
void match(const vector<int>&,int);
bool cmp(int,int);
map<int,int>ch[maxn];
int f[maxn],q[maxn],val[maxn],d[maxn],cnt=0;
int dfn[maxn],size[maxn],son[maxn],top[maxn],sum[maxn]={0},a[maxn];
vector<int>str1[maxn],str2[maxn];
int n,m,s[maxn],len1[maxn],len2[maxn],iter[maxn],ans[maxn];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&len1[i]);
for(int j=0;j<len1[i];j++)scanf("%d",&s[j]);
str1[i]=vector<int>(s,s+len1[i]);
scanf("%d",&len2[i]);
for(int j=0;j<len2[i];j++)scanf("%d",&s[j]);
str2[i]=vector<int>(s,s+len2[i]);
}
for(int i=1,len;i<=m;i++){
scanf("%d",&len);
for(int j=0;j<len;j++)scanf("%d",&s[j]);
iter[i]=insert(s,len);
}
getfail();
bfs();
for(int i=1;i<=n;i++){
a[0]=0;
match(str1[i],len1[i]);
match(str2[i],len2[i]);
sort(a+1,a+a[0]+1,cmp);
sum[a[1]]++;
ans[i]+=val[a[1]];
for(int j=2,x;j<=a[0];j++){
sum[a[j]]++;
ans[i]+=val[a[j]];
x=LCA(a[j],a[j-1]);
sum[x]--;
ans[i]-=val[x];
}
}
for(int i=cnt;i;i--)sum[f[q[i]]]+=sum[q[i]];
for(int i=1;i<=m;i++)printf("%d\n",sum[iter[i]]);
for(int i=1;i<=n;i++){
if(i>1)printf(" ");
printf("%d",ans[i]);
}
return 0;
}
int insert(const int *a,int n){
int x=0;
for(int i=0;i<n;i++){
if(!ch[x].count(a[i]))ch[x][a[i]]=++cnt;
x=ch[x][a[i]];
}
val[x]++;
return x;
}
void getfail(){
int x,head=1,tail=1;
for(map<int,int>::iterator it=ch[x].begin();it!=ch[x].end();it++)q[tail++]=it->second;
d[0]=1;
while(head!=tail){
x=q[head++];
d[x]=d[f[x]]+1;
val[x]+=val[f[x]];
for(map<int,int>::iterator it=ch[x].begin();it!=ch[x].end();it++){
int y=f[x];
while(y&&!ch[y].count(it->first))y=f[y];
f[it->second]=(ch[y].count(it->first)?ch[y][it->first]:0);
q[tail++]=it->second;
}
}
}
void bfs(){
fill(son,son+cnt+1,cnt+1);
for(int i=cnt,x;i;i--){
x=q[i];
dfn[x]=size[f[x]];
size[x]++;
size[f[x]]+=size[x];
if(size[x]>size[son[f[x]]])son[f[x]]=x;
}
for(int i=1,x;i<=cnt;i++){
x=q[i];
dfn[x]+=dfn[f[x]]+1;
if(x==son[f[x]])top[x]=top[f[x]];
else top[x]=x;
}
}
int LCA(int x,int y){
while(top[x]!=top[y]){
if(d[top[x]]<d[top[y]])swap(x,y);
x=f[top[x]];
}
return d[x]<d[y]?x:y;
}
void match(const vector<int>&str,int n){
int x=0;
for(int i=0;i<n;i++){
while(x&&!ch[x].count(str[i]))x=f[x];
if(ch[x].count(str[i]))x=ch[x][str[i]];
a[++a[0]]=x;
}
}
bool cmp(int x,int y){return dfn[x]<dfn[y];}