P2336 [SCOI2012]喵星球上的点名

题目描述

a180285 幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。

假设课堂上有 \(N\) 个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择 \(M\) 个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。

然而，由于喵星人的字码过于古怪，以至于不能用 ASCII 码来表示。为了方便描述，a180285 决定用数串来表示喵星人的名字。

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

输入输出格式

输入格式：

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

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

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

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

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

输出格式：

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

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

说明

\(n\le 5\times 10^4,m\le 10^5\)，串总长\(\le 10^5\)

先建广义SAM

发现点名是子树问题，转换到dfs序

第一问就是区间求颜色数，第二问是统计每种颜色被多少区间覆盖。

用莫队就可以了，在新加入颜色和删去颜色的时候处理一下就行了。

Code:

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <map>
const int N=2e5+10;
std::map <int,int> ch[N];
int col[N],par[N],len[N],las=1,tot=1,id;
void extend(int c)
{int now=++tot,p=las;len[now]=len[p]+1,col[now]=id;while(p&&!ch[p][c]) ch[p][c]=now,p=par[p];if(!p) par[now]=1;else{int x=ch[p][c];if(len[x]==len[p]+1) par[now]=x;else{int y=++tot;len[y]=len[p]+1,par[y]=par[x],col[y]=id,ch[y]=ch[x];while(p&&ch[p][c]==x) ch[p][c]=y,p=par[p];par[now]=par[x]=y;}}las=now;
}
int head[N],to[N],Next[N],cnt;
void add(int u,int v)
{to[++cnt]=v,Next[cnt]=head[u],head[u]=cnt;
}
int dfn[N],low[N],ha[N],dfsclock,n,m,m_;
void dfs(int now)
{ha[dfn[now]=++dfsclock]=now;for(int i=head[now];i;i=Next[i]) dfs(to[i]);low[now]=dfsclock;
}
struct node
{int id,l,r,fi;node(){}node(int id,int l,int r,int fi){this->id=id,this->l=l,this->r=r,this->fi=fi;}bool friend operator <(node a,node b){return a.fi==b.fi?a.l<b.l:a.fi<b.fi;}
}q[N];
int ans[N],sum[N],st[N],tax[N],su;
void add(int p)
{int c=col[ha[p]];if(!tax[c]) ++su,st[c]=id;++tax[c];
}
void del(int p)
{int c=col[ha[p]];--tax[c];if(!tax[c]) --su,sum[c]+=id-st[c];
}
int main()
{scanf("%d%d",&n,&m);for(id=1;id<=n;id++){int Len;las=1;scanf("%d",&Len);for(int c,i=1;i<=Len;i++) scanf("%d",&c),extend(c);las=1;scanf("%d",&Len);for(int c,i=1;i<=Len;i++) scanf("%d",&c),extend(c);}for(int i=2;i<=tot;i++) add(par[i],i);dfs(1);int B=sqrt(dfsclock)+1;for(int i=1;i<=m;i++){int Len,now=1;scanf("%d",&Len);for(int c,j=1;j<=Len;j++){scanf("%d",&c);now=ch[now][c];}if(now) q[++m_]=node(i,dfn[now],low[now],low[now]/B);}std::sort(q+1,q+1+m_);int l=q[1].l,r=l-1;for(id=1;id<=m_;id++){while(q[id].l<l) add(--l);while(q[id].l>l) del(l++);while(q[id].r<r) del(r--);while(q[id].r>r) add(++r);ans[q[id].id]=su;}while(l<=r) del(l++);for(int i=1;i<=m;i++) printf("%d\n",ans[i]);for(int i=1;i<=n;i++) printf("%d ",sum[i]);return 0;
}

2019.1.10