SCOI2012——喵星人的点名（后缀数组）

喵星球上的点名(Large Input)

Time Limit: 2000 ms Memory Limit: 131072 kB Solved: 12 Tried: 58

Submit Status Best Solution Back

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

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

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

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

Input
现在定义喵星球上的字符串给定方法:
先给出一个正整数 L,表示字符串的长度,接下来 L 个整数表示字符串的每个字符。
输入的第一行是两个整数 N 和 M。
接下来有 N 行,每行包含第 i 个喵星人的姓和名两个串。姓和名都是标准的喵星球上的字符串。
接下来有 M 行,每行包含一个喵星球上的字符串,表示老师点名的串。

1 <= N <= 20000 , 1 <= M <= 50000 喵星人的名字总长和点名串的总长分别不超过100000
保证喵星人的字符串中作为字符存在的数不超过 10000
Output
对于每个老师点名的串输出有多少个喵星人应该答到。
然后在最后一行输出每个喵星人被点到多少次。
Sample Input
2 3
6 8 25 0 24 14 8 6 18 0 10 20 24 0
7 14 17 8 7 0 17 0 5 8 25 0 24 0
4 8 25 0 24
4 7 0 17 0
4 17 0 8 25

Sample Output
2
1
0
12

Hint
事实上样例给出的数据如果翻译成地球上的语言可以这样来看
2 3
izayoi sakuya
orihara izaya
izay
hara
raiz
Source

SCOI2012

解析：

         scoi2012的题，也是当年进省队的关键。。。

         其实本题涉及后缀数组部分还比较简单。。。关键是判重，因为一个点名串可能在一个人的名字里出现多次。。。

         我是直接用set，但是只能过lydsy的数据，可能是那个数据比较弱吧。。。

         不管了。。暂时先这样。。。

代码：

/**************************************************************
    Problem: 2754
    User: jianing
    Language: C++
    Result: Accepted
    Time:5900 ms
    Memory:28936 kb
****************************************************************/
 
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<set>
using namespace std;
const int N=600010;
 
int wa[N],ws[N],wb[N],wv[N],sa[N];
int name[N];
int len=-1;
int t1,t2,mid;
int ask[N];
int st[N];
int belong[N];
int ans[N];
 
int cmp(int *s,int a,int b,int l)
{
    return(s[a]==s[b] && s[a+l]==s[b+l]);
}
void da(int *s,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0;i<m;i++)ws[i]=0;
    for(i=0;i<n;i++)ws[x[i]=s[i]]++;
    for(i=1;i<m;i++)ws[i]+=ws[i-1];
    for(i=n-1;i>=0;i--)sa[--ws[x[i]]]=i;
    for(int j=1,p=1;p<n;j*=2,m=p)  //j:当前倍增出的串的长度
    {
        for(p=0,i=n-j;i<n;i++)y[p++]=i;
        for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(i=0;i<n;i++)wv[i]=x[y[i]];
        for(i=0;i<m;i++)ws[i]=0;
        for(i=0;i<n;i++)ws[wv[i]]++;
        for(i=1;i<m;i++)ws[i]+=ws[i-1];
        for(i=n-1;i>=0;i--)sa[--ws[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
 
int rank[N],height[N];
void calheight(int *s,int *sa,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++)rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
        for(k?k--:0,j=sa[rank[i]-1];s[i+k]==s[j+k];k++);
}
 
void read()
{
   // freopen("name.in","r",stdin);
   // freopen("name.out","w",stdout);
    memset(belong,0,sizeof(belong));
    scanf("%d%d",&t1,&t2);
    for(int i=1;i<=t1;i++)
    {
        int x;
        scanf("%d",&x);
        for(int j=1;j<=x;j++)
        {
            int op;
            scanf("%d",&op);
            name[++len]=op+1;
            belong[len]=i;
        }
        name[++len]=0;
        scanf("%d",&x);
        for(int j=1;j<=x;j++)
        {
            int op;
            scanf("%d",&op);
            name[++len]=op+1;
            belong[len]=i;
        }
        name[++len]=0;
    }
    mid=len-1;
    for(int i=1;i<=t2;i++)
    {
        scanf("%d",&ask[i]);
        st[i]=len+1;
        for(int j=1;j<=ask[i];j++)
        {
            int op;
            scanf("%d",&op);
            name[++len]=op+1;
            belong[len]=0;
        }
        name[++len]=0;
    }
   // for(int i=0;i<=len;i++)printf("%d ",name[i]);
}
 
set<int>q;
set<int>::iterator it;
 
void work()
{
    da(name,sa,len+1,10010);
    calheight(name,sa,len);
    for(int i=1;i<=t2;i++)
    {
        int state=rank[st[i]];
        q.clear();
        for(int j=state+1;j<=len;j++)
        {
            if(height[j]<ask[i])break;
            if(sa[j]<=mid)q.insert(belong[sa[j]]);
        }
        for(int j=state-1;j>0;j--)
        {
            if(height[j+1]<ask[i])break;
            if(sa[j]<=mid)q.insert(belong[sa[j]]);
        }
        for(it=q.begin();it!=q.end();++it)++ans[*it];
        printf("%d\n",q.size());
    }
    for(int i=1;i<t1;i++)printf("%d ",ans[i]);
    printf("%d",ans[t1]);
}
 
int main()
{
    read();
    work();
    return 0;
}