[arc064f]Rotated Palindromes

题目大意

有多少字符集为k的长度为n的字符串经过旋转可以变成回文串。

做法

设f[i]表示周期为i的回文串数量。
有$f[i]=k^{(i+1)/2}-\sum_{j|i且j<i}f[j]$
显然border为i的串能旋转出i个字符串。
然后注意一个偶border串会被算两次。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int mo=1000000007,inv2=(mo+1)/2;
int d[3000],f[3000];
int i,j,k,l,t,n,m,top,ans;
int qsm(int x,int y){
    if (!y) return 1;
    int t=qsm(x,y/2);
    t=(ll)t*t%mo;
    if (y%2) t=(ll)t*x%mo;
    return t;
}
int main(){
    scanf("%d%d",&n,&k);
    t=floor(sqrt(n));
    fo(i,1,t)
        if (n%i==0){
            d[++top]=i;
            if (i!=n/i) d[++top]=n/i;
        }
    sort(d+1,d+top+1);
    fo(i,1,top){
        f[i]=qsm(k,(d[i]+1)/2);
        fo(j,1,i-1)
            if (d[i]%d[j]==0) (f[i]-=f[j])%=mo;
        if (d[i]%2==0) (ans+=(ll)f[i]*d[i]%mo*inv2%mo)%=mo;
        else (ans+=(ll)f[i]*d[i]%mo)%=mo;
    }
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}