【bzoj2160】拉拉队排练 manacher

先用manacher求出数组后,用一个差分数组计算每一个长度的字符串出现的次数,暴力计算就好了。


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 1000100
#define mod 19930726

using namespace std;

long long sum[maxn];
char s[maxn];
int p[maxn];
int n,m;
long long k,ans,cnt;

long long power(long long x,long long y)
{
	long long ans=1;
	while (y)
	{
		if (y&1) ans=ans*x%mod;
		x=x*x%mod;
		y>>=1;
	}
	return ans;
}

int main()
{
	scanf("%d%lld",&n,&k);
	scanf("%s",s+1);
	int mx=0,id=0;s[0]='#';s[n+1]='!';
	for (int i=1;i<=n;i++)
	{
		if (mx>i) p[i]=min(p[2*id-i],mx-i); else p[i]=1;
		while (s[i+p[i]]==s[i-p[i]]) p[i]++;
		if (i+p[i]>mx) mx=i+p[i],id=i;
		sum[0]++;sum[p[i]+1]--;
	}
	for (int i=1;i<=n;i++) sum[i]+=sum[i-1];
	ans=1;cnt=0;
	for (int i=n;i>=1;i--)
	  if (cnt+sum[i]<k) ans=ans*power(2*i-1,sum[i])%mod,cnt+=sum[i];
	  else {ans=ans*power(2*i-1,k-cnt)%mod;cnt=k;break;}
	if (cnt<k) printf("-1\n");
	else printf("%lld\n",ans);
	return 0;
}


你可能感兴趣的:(【bzoj2160】拉拉队排练 manacher)