2754: [SCOI2012]喵星球上的点名
Description
a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。 假设课堂上有N个喵星人，每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名，每次读出一个串的时候，如果这个串是一个喵星人的姓或名的子串，那么这个喵星人就必须答到。 然而，由于喵星人的字码过于古怪，以至于不能用ASCII码来表示。为了方便描述，a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到，以及M次点名结束后每个喵星人答到多少次吗？
Input
现在定义喵星球上的字符串给定方法：
先给出一个正整数L，表示字符串的长度，接下来L个整数表示字符串的每个字符。
输入的第一行是两个整数N和M。
接下来有N行，每行包含第i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的字符串。
接下来有M行，每行包含一个喵星球上的字符串，表示老师点名的串。
Output
对于每个老师点名的串输出有多少个喵星人应该答到。
然后在最后一行输出每个喵星人被点到多少次。
把姓名串和询问串连在一起求\(SA\)，以下为\(rk\)意义下；
然后对于每一个询问串二分求出\(LCP\)等于它长度的区间，及询问串出现位置；
这样的话第一问就是求每个区间内不同的数的个数，即\(HH\)的项链，莫队/树状数组；
第二问一个区间中属于同一个串的多个后缀会被重复计算，所以把相邻的重复后缀的贡献作差；
对于每个询问\([l,r]\)，在\(l\)处将\(w\)区间加\(1\)，\(r+1\)处区间减\(1\)，则后缀\(i\)的贡献为\(w[i]-w[las[num[i]]]\)；
\(las[num[i]]\)为上一个与\(i\)母串相同的后缀，树状数组即可；
AC GET☆DAZE
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#define N 600039
#define mod 20070831
#define inf 0x3f3f3f3f
#define ll long long
using namespace std;
struct question
{
int sd,l,r,ans;
}ask[N/6];
int neko,wa,fir[N/3],len[N/3],bel[N],bit[N],ans[N];
int n,m,num[N],SA[N],rk[N],Height[N],las[N],buc[N],pw[20],st[N][20];
bool cmp_hh(question i,question j) {return i.r<j.r;}
bool cmp_sd(question i,question j) {return i.sd<j.sd;}
void get_string(int stp)
{
fir[stp]=n+1;
scanf("%d",&len[stp]);
for(int a=1;a<=len[stp];a++)
{
scanf("%d",&num[++n]);
num[n]++,bel[n]=stp;
}
num[++n]=10001+stp;
}
bool judge(int i,int j,int k)
{
return las[i]==las[j] && las[i+k]==las[j+k];
}
void radix_sort()
{
for(int a=1;a<=m;a++) buc[a]=0;
for(int a=1;a<=n;a++) buc[rk[las[a]]]++;
for(int a=1;a<=m;a++) buc[a]+=buc[a-1];
for(int a=n;a>=1;a--) SA[buc[rk[las[a]]]--]=las[a];
}
void get_SA()
{
for(int a=1;a<=n;a++) rk[a]=num[a],las[a]=a;
m=N-39,radix_sort();
for(int a=1,b=0;m<n || a==1;a<<=1,b=0)
{
for(int c=n-a+1;c<=n;c++) las[++b]=c;
for(int c=1;c<=n;c++) if(SA[c]>a) las[++b]=SA[c]-a;
radix_sort(),m=0;
for(int c=1;c<=n;c++) las[c]=rk[c];
for(int c=1;c<=n;c++) rk[SA[c]]=(judge(SA[c-1],SA[c],a) ? m : ++m);
}
for(int a=1,b=0;a<=n;a++,b=max(0,b-1))
{
while(num[a+b]==num[SA[rk[a]-1]+b]) b++;
Height[rk[a]]=b;
}
}
void get_st()
{
pw[0]=1;
for(int a=1;pw[a-1]<=n;a++) pw[a]=pw[a-1]<<1;
for(int a=1;a<=n;a++) st[a][0]=Height[a];
for(int a=1;pw[a]<=n;a++)
{
for(int b=1;b<=n;b++)
{
st[b][a]=min(st[b][a-1],st[b+pw[a-1]][a-1]);
}
}
}
int RMQ(int l,int r)
{
int stp=log((double)r-l+1)/log(2.0);
return min(st[l][stp],st[r-pw[stp]+1][stp]);
}
int search_mae(int stp)
{
int l=1,r=rk[fir[stp]]-1,mid,res=rk[fir[stp]];
while(l<=r)
{
mid=l+r>>1;
if(RMQ(mid+1,rk[fir[stp]])<len[stp]) l=mid+1;
else r=mid-1,res=mid;
}
return res;
}
int search_ato(int stp)
{
int l=rk[fir[stp]]+1,r=n,mid,res=rk[fir[stp]];
while(l<=r)
{
mid=l+r>>1;
if(RMQ(rk[fir[stp]]+1,mid)<len[stp]) r=mid-1;
else l=mid+1,res=mid;
}
return res;
}
void update(int p,int v)
{
if(!p) return;
while(p<=n) bit[p]+=v,p+=(-p&p);
}
int query(int p)
{
int res=0;
while(p) res+=bit[p],p-=(-p&p);
return res;
}
int main()
{
scanf("%d%d",&neko,&wa);
for(int a=1,b;a<=neko;a++)
{
get_string(2*a-1);
get_string(2*a);
}
for(int a=1,b;a<=wa;a++)
{
get_string(2*neko+a);
}
get_SA(),get_st();
for(int a=1;a<=wa;a++)
{
ask[a].sd=a;
ask[a].l=search_mae(2*neko+a);
ask[a].r=search_ato(2*neko+a);
}
sort(ask+1,ask+wa+1,cmp_hh);
for(int a=0;a<=neko;a++) las[a]=0;
for(int a=1,b=1;a<=wa;a++)
{
while(b<=ask[a].r)
{
if(bel[SA[b]]<=2*neko)
{
update(las[bel[SA[b]]+1>>1],-1);
las[bel[SA[b]]+1>>1]=b;
update(b,1);
}
b++;
}
ask[a].ans=query(ask[a].r)-query(ask[a].l-1);
}
for(int a=0;a<=neko;a++) las[a]=0;
for(int a=1;a<=n;a++) bit[a]=0;
for(int a=1;a<=wa;a++) update(ask[a].l,1);
for(int a=1,b=1;a<=n;a++)
{
while(b<=wa && ask[b].r<a)
{
update(ask[b].l,-1);
b++;
}
if(bel[SA[a]]<=2*neko)
{
ans[bel[SA[a]]+1>>1]+=query(a)-query(las[bel[SA[a]]+1>>1]);
las[bel[SA[a]]+1>>1]=a;
}
}
sort(ask+1,ask+wa+1,cmp_sd);
for(int a=1;a<=wa;a++) printf("%d\n",ask[a].ans);
for(int a=1;a<=neko;a++)
{
printf("%d",ans[a]);
if(a<neko) printf(" ");
}
return 0;
}