首页 > 代码库 > BZOJ2754: [SCOI2012]喵星球上的点名

BZOJ2754: [SCOI2012]喵星球上的点名

2754: [SCOI2012]喵星球上的点名

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 680  Solved: 314
[Submit][Status]

Description

a180285幸运地被选做了地球到喵星球的留学生。他发现喵星人在上课前的点名现象非常有趣。   假设课堂上有N个喵星人,每个喵星人的名字由姓和名构成。喵星球上的老师会选择M个串来点名,每次读出一个串的时候,如果这个串是一个喵星人的姓或名的子串,那么这个喵星人就必须答到。 然而,由于喵星人的字码过于古怪,以至于不能用ASCII码来表示。为了方便描述,a180285决定用数串来表示喵星人的名字。
现在你能帮助a180285统计每次点名的时候有多少喵星人答到,以及M次点名结束后每个喵星人答到多少次吗?  

Input

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

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
1 2
【提示】
事实上样例给出的数据如果翻译成地球上的语言可以这样来看
2 3
izayoi sakuya
orihara izaya
izay
hara
raiz

HINT

 



【数据范围】 

 对于30%的数据,保证: 

1<=N,M<=1000,喵星人的名字总长不超过4000,点名串的总长不超过2000。

对于100%的数据,保证:

1<=N<=20000,1<=M<=50000,喵星人的名字总长和点名串的总长分别不超过100000,保证喵星人的字符串中作为字符存在的数不超过10000。

 

Source

题解:

没想到后缀数组的解法这么暴力也能过。。。

把姓+名+询问全部放到一个字符串里做后缀数组,对每一个询问向前、向后扫描并把答案放入set,set判重并输出size()

一开始好像是中间填的字符越界了,导致一直WA了好几天。。。

本以为我对字符串重新编一下号就能使m减小而程序跑得更快,没想到还是慢成翔啊。。。

代码:

  1 #include<cstdio>  2    3 #include<cstdlib>  4    5 #include<cmath>  6    7 #include<cstring>  8    9 #include<algorithm> 10   11 #include<iostream> 12   13 #include<vector> 14   15 #include<map> 16   17 #include<set> 18   19 #include<queue> 20   21 #include<string> 22   23 #define inf 1000000000 24   25 #define maxn 1000000+1000 26   27 #define maxm 500+100 28   29 #define eps 1e-10 30   31 #define ll long long 32   33 #define pa pair<int,int> 34   35 #define for0(i,n) for(int i=0;i<=(n);i++) 36   37 #define for1(i,n) for(int i=1;i<=(n);i++) 38   39 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 40   41 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 42   43 #define mod 1000000007 44   45 using namespace std; 46   47 inline int read() 48   49 { 50   51     int x=0,f=1;char ch=getchar(); 52   53     while(ch<0||ch>9){if(ch==-)f=-1;ch=getchar();} 54   55     while(ch>=0&&ch<=9){x=10*x+ch-0;ch=getchar();} 56   57     return x*f; 58   59 } 60 set<int>s; 61 set<int>::iterator it; 62 int a[maxn],b[maxn],v[maxn],ans[maxn],rank[maxn],h[maxn],sa[maxn],len[maxn],pos[maxn],wr[maxn],wa[maxn],wb[maxn]; 63 int n,m,cnt,tot,mid; 64 inline bool cmp(int *r,int a,int b,int l) 65 { 66     return r[a]==r[b]&&r[a+l]==r[b+l]; 67 } 68 void da(int n,int m) 69 { 70     int *x=wa,*y=wb,*t; 71     for0(i,m)wr[i]=0; 72     for0(i,n)wr[x[i]=a[i]]++; 73     for1(i,m)wr[i]+=wr[i-1]; 74     for3(i,n,0)sa[--wr[x[i]]]=i; 75     for(int p=1,j=1;p<=n;j<<=1,m=p) 76     { 77         p=0; 78         for2(i,n-j+1,n)y[p++]=i; 79         for0(i,n)if(sa[i]>=j)y[p++]=sa[i]-j; 80         for0(i,m)wr[i]=0; 81         for0(i,n)wr[x[y[i]]]++; 82         for1(i,m)wr[i]+=wr[i-1]; 83         for3(i,n,0)sa[--wr[x[y[i]]]]=y[i]; 84         int i; 85         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<=n;i++) 86           x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; 87     } 88 }    89 void geth(int n) 90 { 91     for1(i,n)rank[sa[i]]=i; 92     for(int i=0,k=0,j;i<n;h[rank[i++]]=k) 93      for(k?k--:0,j=sa[rank[i]-1];a[i+k]==a[j+k];k++); 94 } 95   96 int main() 97   98 { 99  100     freopen("input.txt","r",stdin);101  102     freopen("output.txt","w",stdout);103  104     n=read();m=read();tot=-1;105     for1(i,n)106     {107         int x=read();108         for1(j,x)a[++tot]=read()+1,b[tot]=i;109         x=read();110         a[++tot]=0;111         for1(j,x)a[++tot]=read()+1,b[tot]=i;112         a[++tot]=0;113     }114     a[++tot]=0;mid=tot;115     for1(i,m)116     {117         len[i]=read();118         pos[i]=tot+1;119         for1(j,len[i])a[++tot]=read()+1;120         a[++tot]=0;121     }122     cnt=0;123     for0(i,tot)if(!v[a[i]])a[i]=v[a[i]]=++cnt;else a[i]=v[a[i]];124     a[++tot]=0;125     da(tot,cnt);126     geth(tot);127     for1(i,m)128     {129         s.clear();130         int x=rank[pos[i]],y=len[i];131         for2(j,x+1,tot)132          {133              if(h[j]<y)break;134              if(sa[j]>=mid)continue;135              if(b[sa[j]])s.insert(b[sa[j]]);136          }137         for3(j,x,1)138           {139               if(h[j]<y)break;140               if(sa[j-1]>=mid)continue;141               if(b[sa[j-1]])s.insert(b[sa[j-1]]);142           }143         for(it=s.begin();it!=s.end();it++)ans[*it]++;144         printf("%d\n",s.size());145     }146     for1(i,n-1)printf("%d ",ans[i]);printf("%d\n",ans[n]);147  148     return 0;149  150 }
View Code

 

BZOJ2754: [SCOI2012]喵星球上的点名