【BZOJ 2160】 拉拉队排练

【题目链接】

https://www.lydsy.com/JudgeOnline/problem.php?id=2160

【算法】

先简化题意 ： 给定一个字符串，求最长的k个奇回文子串长度的乘积

先运行Manacher算法，对于每个位置i，我们知道以i为中心的回文串的最长半径为pi，那么i这个位置对半径为1-p[i]的回文串的个数都产生了1的”贡献“

因此，我们可以用差分求出任意半径的回文串个数，然后统计答案即可，注意要用快速幂

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 1000010
const int P = ;
typedef long long ll;

int i,len,n;
ll sum,ans,k;
int p[MAXN];
ll cnt[MAXN];
char s[MAXN];

inline ll power(ll a,ll n)
{
        ll ans = ,b = a;
        while (n)
        {
                if (n & ) ans = 1ll * ans * b % P;
                n >>= ;
                b = 1ll * b * b % P;
        }
        return ans;
}

inline void Manacher()
{
        int i,pos = ,mx = ;
        for (i = ; i <= len; i++)
        {
                if (mx > i)    p[i] = min(p[*pos-i],mx-i);
                else p[i] = ;
                while (i - p[i] >=  && i + p[i] <= len && s[i-p[i]] == s[i+p[i]]) p[i]++;
                if (i + p[i] -  > mx)
                {
                        mx = i + p[i] - ;
                        pos = i;
                }
        }
}

int main()
{

        scanf("%d%lld%s",&n,&k,s+);
        len = strlen(s+);
        Manacher();
        for (i = ; i <= len; i++)
        {
                cnt[]++;
                cnt[p[i]+]--;
        }
        for (i = ; i <= (len + ) / ; i++) cnt[i] += cnt[i-];
        sum = ; ans = ;
        for (i = (len + ) / ; i >= ; i--)
        {
                if (cnt[i] <= ) continue;
                if (sum + cnt[i] < k)
                {
                        sum += cnt[i];
                        ans = 1ll * ans * power(*i-,cnt[i]) % P;
                } else
                {
                        ans = 1ll * ans * power(*i-,k-sum) % P;
                        sum += cnt[i];
                        break;
                }
        }
        if (sum < k) printf("-1\n");
        else printf("%lld\n",ans);

        return ;

}