NOI2014 动物园

题目

NOI2014 动物园_第1张图片

题解

#include
using namespace std;
inline int read()
{
    int num=0;
    char c=' ';
    bool flag=true;
    for(;c>'9'||c<'0';c=getchar())
    if(c=='-')
    flag=false;
    for(;c>='0'&&c<='9';num=num*10+c-48,c=getchar());
    return flag ? num : -num;
}
const int maxn=1000020;
const long long mod=1000000007;
string s;
int num[maxn],f[maxn];
//f来记录匹配失败的次数。
//num可以递推求解

void work()
{
    f[0]=-1;f[1]=0;
    num[0]=0;num[1]=1;
    int j=0;
    for(int i=1;iwhile(j>=0&&s[i]!=s[j])
            j=f[j];
        f[i+1]=++j;
        num[i+1]=num[j]+1;
    }//第一遍先求出next数组
    j=0;
    unsigned long long ans=1;
    for(int i=1;iwhile(j>=0&&s[i]!=s[j])
            j=f[j];
        j++;
        while((j<<1)>(i+1))
            j=f[j];
        ans*=(unsigned long long)num[j]+1;
        ans%=mod;
    }//第二遍求解num数组的累乘积
    printf("%lld\n",ans);
}
int main()
{
    int _=read();
    while(_--)//多组数据
    {
        cin>>s;//字符串
        work();
    }
    return 0;
}

你可能感兴趣的:(题解,字符串)