后缀自动机五·重复旋律8
时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
小Hi平时的一大兴趣爱好就是演奏钢琴。我们知道一段音乐旋律可以被表示为一段数构成的数列。
小Hi发现旋律可以循环，每次把一段旋律里面最前面一个音换到最后面就成为了原旋律的“循环相似旋律”，还可以对“循环相似旋律”进行相同的变换能继续得到原串的“循环相似旋律”。
小Hi对此产生了浓厚的兴趣，他有若干段旋律，和一部音乐作品。对于每一段旋律，他想知道有多少在音乐作品中的子串（重复便多次计）和该旋律是“循环相似旋律”。
输入
第一行，一个由小写字母构成的字符串S，表示一部音乐作品。字符串S长度不超过100000。
第二行，一个整数N，表示有N段旋律。接下来N行，每行包含一个由小写字母构成的字符串str，表示一段旋律。所有旋律的长度和不超过 100000。
输出
输出共N行，每行一个整数，表示答案。
- 样例输入
-
abac
3
a
ab
ca - 样例输出
-
2
2
1#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <time.h>
#include <string>
#include <map>
#include <stack>
#include <vector>
#include <set>
#include <queue>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=4e6+;
const int M=1e6+;
const long long MOD = 1000000007LL;
int tot,slink[*N],trans[*N][],minlen[*N],maxlen[*N],edpts[*N];
string str;
int n,sz_end[N*];
int containPrefix[N],ind[N],ans[*N+],cnt[N*],green[*N];
ll sum[N*],sz_valid[N*];
int newstate(int _maxlen,int _minlen,int* _trans,int _slink,int _green) {
maxlen[++tot]=_maxlen;
minlen[tot]=_minlen;
slink[tot]=_slink;
green[tot]=_green;
if(_trans)
for(int i=; i<; i++)
trans[tot][i]=_trans[i];
return tot;
}
int add_char(char ch,int u) {
int c=ch-'a',v=u;
int z=newstate(maxlen[u]+,-,NULL,,);
containPrefix[z]=;
while(v&&!trans[v][c]) {
trans[v][c]=z;
v=slink[v];
}
if(!v) {
minlen[z]=;
slink[z]=;
ind[]++;
return z;
}
int x=trans[v][c];
if(maxlen[v]+==maxlen[x]) {
slink[z]=x;
minlen[z]=maxlen[x]+;
ind[x]++;
return z;
}
int y=newstate(maxlen[v]+,-,trans[x],slink[x],);
slink[z]=slink[x]=y;
ind[y]+=;
minlen[x]=minlen[z]=maxlen[y]+;
while(v&&trans[v][c]==x) {
trans[v][c]=y;
v=slink[v];
}
minlen[y]=maxlen[slink[y]]+;
return z;
}
void init_dag() {
for ( int i = ; i <=tot; i++ ) {
if(slink[i]>)cnt[slink[i]]++;
}
} void topsort() {
queue<int>q;
for ( int i = ; i <=tot; i++ ) {
if ( cnt[i] == ) {
q.push(i);
sz_valid[i] = ;
sum[i] = ;
}
}
while(!q.empty()) {
int i=q.front();
q.pop();
if(green[i])sz_end[i]++;
int j=slink[i];
sz_end[j]+=sz_end[i];
if(!(--cnt[j]))q.push(j);
}
}
int solve() {
vector<bool> visited(N*,false);
int res = ;
string t;
cin>>t;
int len = t.size();
for(int i = ; i < len-; i++) {
t+= t[i];
}
int len2 = len*-;
int u,l;
u = ;
l = ;
for(int i = ; i < len2; i++) {
int c = t[i] - 'a';
for(; u!= && trans[u][c]==;) {
u = slink[u];
l = maxlen[u];
}
if(trans[u][c] >) {
u = trans[u][c];
l = l + ;
} else {
u = ;
l = ;
}
if(l > len) {
for(; maxlen[slink[u]] >= len;) {
u = slink[u];
l = maxlen[u];
}
}
if(l >= len && !visited[u]) {
visited[u] = true;
res += sz_end[u];
}
}
return res;
}
int main() {
cin>>str;
cin>>n;
int pre=;
tot=;
int len=str.length();
for(int i=; i<len; i++) {
pre=add_char(str[i],pre);
}
init_dag();
topsort();
for(int i=;i<n;i++){
printf("%d\n",solve());
}
return ;
}