bzoj5130: [Lydsy1712月赛]字符串的周期

这道题很有意思啊。

字符串循环节用KMP（手推一下）

假如是26^12肯定很不滋磁

但是可以发现ABA和BCB和BAB这些都是等价的

那就把最小的拿出来搞再乘个排列数就好了

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
const LL mod=;

int n,m,a[],p[];LL mi[],ans;
LL KMP()
{
    p[]=;int j=;
    for(int i=;i<=n;i++)
    {
        while(j>&&a[j+]!=a[i])j=p[j];
        if(a[j+]==a[i])j++;
        p[i]=j;
    }

    LL ret=;
    for(int i=;i<=n;i++)
        ret=(ret*(i-p[i]))%mod;
    return ret;
}
void dfs(int k,int maxp)
{
    if(k==n+)
    {
        ans=(ans+mi[maxp]*KMP())%mod;
        return ;
    }
    for(int i=;i<=maxp;i++)
    {
        a[k]=i;
        dfs(k+,maxp);
        a[k]=;
    }
    a[k]=++maxp;dfs(k+,maxp);
}
int main()
{
    scanf("%d%d",&n,&m);
    mi[]=;for(int i=;i<=n;i++)mi[i]=(mi[i-]*(m-i+))%mod;
    ans=;dfs(,);
    printf("%lld\n",ans);
    return ;
}