Atcoder Regular Contest 064 F Rotated Palindromes

Rotated Palindromes

Problem Statement

给出$n$,$k$。
问有多少个长度为$n$的序列$a$，对于任意$1$≤$i$≤$n$满足$1$≤$a_i$≤$k$,并且序列$a$经过若干次循环左移能变成回文序列.
$For$ $example$:循环左移一次$1,1,2,3$ -> $1,2,3,1$
$1$≤$n$,$k$≤$10^9$

Solution

给出一个显然的结论，一个回文串的最小循环节一定回文。
先找到$n$的所有约数，设第$i$个约数为$X_i$。
设$f_i$表示序列的最小循环节长度为$X_i$，最小循环节的个数。
考虑$f_i$对答案的贡献。
循环节对应的序列左移操作$X_i$次，就可以得到$X_i$种不同的序列。
但可以发现，若$X_i$为偶数时会算重一半，乘上$1\over 2$。
$f_i$的求法也不难，容斥即可。即$f_i$=$k^{\lfloor{{X_i+1}\over 2}\rfloor}$-$\sum_{X_d|X_i}f_d$
时间复杂度$O$($σ_0^2$$(n)$)

Code

#include<bits/stdc++.h>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll N=2e5,mo=1e9+7;

int n,o,p;
ll f[N],a[N],k;

ll ksm(ll o,ll t)
{
    ll y=1;
    for(;t;t>>=1,o=o*o%mo)
    if(t&1)y=y*o%mo;
    return y;
}

int main()
{
    cin>>n>>k;
    for(int i=1;i*i<=n;++i)
    if(n%i==0){
        a[++o]=i;
        if(i*i%n)a[++o]=n/i;
    }
    sort(a+1,a+o+1); 
    ll ans=0;
    fo(i,1,o){
        f[i]=ksm(k,(a[i]+1)>>1);
        fo(l,1,i-1)if(a[i]%a[l]==0)f[i]=(f[i]-f[l]+mo)%mo;
        (a[i]&1)?ans=(ans+f[i]*a[i])%mo:ans=(ans+(f[i]*a[i]>>1))%mo;
    }
    cout<<ans;
}