[arc064f]Rotated Palindromes

题目大意

给你n,k。
构造一个数组a[1..n]。
满足他是一个回文串，且a[i]取值为1..K。
然后你可以把他旋转任意次，旋转一次就比如aaab变成aaba。设旋转后的数组为a’，问有多少个a’，对1e9+7取模。
n,K<=1e9

解题思路

考虑怎么才会重复。
如果a[]有一个per的周期，而且per|n，那么a[]旋转若干次后才会重复。
所以一个这样的a[]会贡献per个a’？漏了一种重复情况。如果2|per，你旋转per/2次之后，会变成一个per更大的回文串。所以这种情况只有per/2的贡献。
设f[x]表示per为n的第x大个约数时，有多少个a[]。那么你容斥去一下重就行。
然后考虑每个f[x]对ans的贡献，乘上per或per/2即可。
因数不多，时间复杂度O(d(n)^2)可做。

代码

#include<cstdio> 
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
#define cmax(a,b) (a=(a>b)?a:b)
#define cmin(a,b) (a=(a<b)?a:b)
typedef long long ll;
const int N=1e6+5,M=2e6+5,mo=1e9+7;
ll f[N],fac[N],i,n,K,j,ans;
ll ksm(ll x,ll y)
{
    ll ret=1;
    while (y)
    {
        if (y&1) ret=ret*x%mo;
        y>>=1;
        x=x*x%mo;
    }
    return ret;
}
void predo()
{
    fo(i,1,trunc(sqrt(n)))
    if (n%i==0)
    {
        fac[++fac[0]]=i;
        if (i*i!=n) fac[++fac[0]]=n/i;
    }
    sort(fac+1,fac+1+fac[0]);
}
int main()
{
    freopen("t14.in","r",stdin);
    //freopen("t14.out","w",stdout);
    scanf("%lld %lld",&n,&K);
    predo();
    fo(i,1,fac[0])
    {
        f[i]=ksm(K,(fac[i]+1)/2);
        fo(j,1,i-1)
            if (fac[i]%fac[j]==0)
                (f[i]-=f[j])%=mo;
        if (fac[i]%2)
            (ans+=f[i]*fac[i])%=mo;
        else (ans+=f[i]*fac[i]/2)%=mo;
    }
    printf("%lld\n",(ans+mo)%mo);
}