[BZOJ2754] [SCOI2012]喵星球上的点名解题报告|后缀数组

时间:2021-02-13 14:40:57
  
  a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。   假设课堂上有N个喵星人,每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名,每次读出一个串的时候,如果这个串是一个喵星人的姓或名的子串,那么这个喵星人就必须答到。 然而,由于喵星人的字码过于古怪,以至于不能用ASCII码来表示。为了方便描述,a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到,以及M次点名结束后每个喵星人答到多少次吗?  
 
  应该算是后缀数组比较简单的应用了吧...
  首先把所有的串都接起来,中间用不同的分隔符隔开
  然后进行后缀数组将sa,rank,height等数组求出来
  最后枚举每一个点名串,以rank[点名串起点]开始向左和向右拓展
  这里的拓展是指在sa的下标里拓展
  边统计边累计每个喵星人的点到次数
  为了避免一次点名中统计到一个喵星人多次,以及一个喵星人一次点名却统计了多次
  (实际上二者是一样的...)
  都可以用一个标号数组来处理
  然而这个标号数组如果每次fillchar显然不现实...因为太慢...
  然后就运用到KM算法里顶标更新的思想
  当然,再做一次然后把数组还原也是可以的
 
  另外在BZ上PE了一发...行末注意不能留有空格
 program bzoj2754;
const INF = ;maxn = ;
var tota,totb,i,cnt,n,x,j:longint;
a,rank,s,tmp,sa,height,pos,ter,len,vis,ans:array[-..maxn]of longint; function max(a,b:longint):longint;
begin
if a>b then exit(a) else exit(b);
end; procedure Suffix_Array;
var sz,i,j,p,v0,v1,v00,v01:longint;
begin
sz:=max(n,INF+);
for i:= to sz do s[i]:=;
for i:= to n- do rank[i]:=a[i];
for i:= to n- do inc(s[rank[i]]);
for i:= to sz do inc(s[i],s[i-]);
for i:=n- downto do
begin
dec(s[rank[i]]);
sa[s[rank[i]]]:=i;
end;
j:=;
while j<=n do
begin
p:=;
for i:=n-j to n- do
begin
tmp[p]:=i;inc(p);
end;
for i:= to n- do if sa[i]-j>= then
begin
tmp[p]:=sa[i]-j;inc(p);
end;
for i:= to sz do s[i]:=;
for i:= to n- do inc(s[rank[i]]);
for i:= to sz do inc(s[i],s[i-]);
for i:=n- downto do
begin
dec(s[rank[tmp[i]]]);
sa[s[rank[tmp[i]]]]:=tmp[i];
end;
p:=;tmp[sa[]]:=;
for i:= to n- do
begin
v0:=sa[i-];v1:=sa[i];
if v0+j<n then v00:=rank[v0+j] else v00:=-;
if v1+j<n then v01:=rank[v1+j] else v01:=-;
if (rank[v0]=rank[v1])and(v00=v01) then tmp[sa[i]]:=p else
begin
inc(p);tmp[sa[i]]:=p;
end;
end;
for i:= to n- do rank[i]:=tmp[i];
j:=j << ;
end;
end; function compare(i,j,x:longint):longint;
begin
while (i+x-<n)and(j+x-<n)and(a[i+x-]=a[j+x-]) do inc(x);
exit(x-);
end; procedure calc_height;
var i:longint;
begin
if rank[]= then height[]:= else height[]:=compare(,sa[rank[]-],);
for i:= to n- do
if rank[i]= then height[i]:= else height[i]:=compare(i,sa[rank[i]-],max(height[i-],));
end; function solve(x,y:longint):longint;
var i,tot:longint;
begin
x:=rank[x];i:=x;tot:=;
while (i>=)and(height[sa[i]]>=y) do
begin
if (pos[sa[i-]]<>)and(vis[pos[sa[i-]]]<>x) then
begin
inc(tot);vis[pos[sa[i-]]]:=x;inc(ans[pos[sa[i-]]]);
end;
dec(i);
end;
i:=x+;
while (i<n)and(height[sa[i]]>=y) do
begin
if (pos[sa[i]]<>)and(vis[pos[sa[i]]]<>x) then
begin
inc(tot);vis[pos[sa[i]]]:=x;inc(ans[pos[sa[i]]]);
end;
inc(i);
end;
exit(tot);
end; begin
readln(tota,totb);cnt:=;n:=-;
for i:= to tota do
begin
read(x);for j:= to x do begin inc(n);read(a[n]);inc(a[n],INF);pos[n]:=i;end;
inc(cnt);inc(n);a[n]:=cnt;
read(x);for j:= to x do begin inc(n);read(a[n]);inc(a[n],INF);pos[n]:=i;end;
inc(cnt);inc(n);a[n]:=cnt;
end;
for i:= to totb do
begin
ter[i]:=n+;
read(x);for j:= to x do begin inc(n);read(a[n]);inc(a[n],INF);end;
len[i]:=x;
inc(cnt);inc(n);a[n]:=cnt;
end;
fillchar(vis,sizeof(vis),);
inc(n);
suffix_array; calc_height;
fillchar(ans,sizeof(ans),);
for i:= to totb do writeln(solve(ter[i],len[i]));
for i:= to tota- do write(ans[i],' ');writeln(ans[tota]);
end.