hihocoder #1465 : 后缀自动机五·重复旋律8

 

#1465 : 后缀自动机五·重复旋律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<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
//by NeighThorn
using namespace std;

const int maxn=300000+5;

int n,last=1,tail=1,cnt[maxn],out[maxn],vis[maxn],Max[maxn],Min[maxn],nxt[maxn][26],fail[maxn];

char s[maxn];

inline void build(char *s){
	while(*s){
		int p=last,t=++tail,c=*s++-'a';
		Max[t]=Max[p]+1;cnt[t]=1;
		while(p&&!nxt[p][c])
			nxt[p][c]=t,p=fail[p];
		if(p){
			int q=nxt[p][c];
			if(Max[q]==Max[p]+1)
				Min[t]=Max[q]+1,fail[t]=q;
			else{
				int k=++tail;
				fail[k]=fail[q];
				fail[q]=fail[t]=k;
				Max[k]=Max[p]+1;
				Min[q]=Max[k]+1;
				Min[t]=Max[k]+1;
				Min[k]=Max[fail[k]]+1;
				memcpy(nxt[k],nxt[q],26*sizeof(int));
				while(p&&nxt[p][c]==q)
					nxt[p][c]=k,p=fail[p];
			}
		}
		else
			Min[t]=fail[t]=1;
		last=t;
	}
}

inline void calc(void){
	int hd=1,ta=0,q[maxn];
	for(int i=1;i<=tail;i++)
		out[fail[i]]++;
	for(int i=1;i<=tail;i++)
		if(!out[i])
			q[++ta]=i;
	while(hd<=ta){
		int top=q[hd++];
		cnt[fail[top]]+=cnt[top];
		out[fail[top]]--;
		if(!out[fail[top]])
			q[++ta]=fail[top];
	}
}

inline void solve(char *s){
	memset(vis,0,sizeof(vis));
	int u=1,n=strlen(s),len=0,ans=0;
	for(int i=n,j=0;j<n-1;j++,i++) s[i]=s[j];
	s[n+n-1]='\0';
	while(*s){
		int c=*s++-'a';
		while(u!=1&&!nxt[u][c])
			u=fail[u],len=Max[u];
		if(nxt[u][c])
			u=nxt[u][c],len++;
		else
			u=1,len=0;
		if(len>n)
			while(Max[fail[u]]>=n)
				u=fail[u],len=Max[u];
		if(len>=n&&!vis[u])
			vis[u]=1,ans+=cnt[u];
	}
	printf("%d\n",ans);
}

signed main(void){
	scanf("%s",s);build(s);calc();
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%s",s),solve(s); 
	return 0;
}

　　

$By$  $NeighThorn$

posted @ 2017-04-20 09:07  NeighThorn  阅读(373)  评论(0)  编辑  收藏  举报