uva12387 - Alphabet Soup

链接

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3809

题解

要让旋转之后重合，确定下第一个点转到哪里之后，剩下的只需要相对次序不变就好，那么我就把这个序列先复制一遍放在末尾，然后差分，对原序列也差分，做一个匹配就好了，我用的是kmp
计数的任务交给polya定理

代码

//polya定理、kmp
#include <bits/stdc++.h>
#define mod 100000007ll
#define maxn 1000010
#define cl(x) memset(x,0,sizeof(x))
using namespace std;
typedef long long ll;
ll fastpow(ll a, ll b)
{
	ll t=a, ans=1;
	for(;b;b>>=1,t=t*t%mod)if(b&1)ans=ans*t%mod;
	return ans;
}
ll gcd(ll a, ll b){return !b?a:gcd(b,a%b);}
ll read(ll x=0)
{
	ll c, f=1;
	for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
	for(;isdigit(c);c=getchar())x=x*10+c-48;
	return f*x;
}
ll nex[maxn], a[maxn], S, P, t[maxn];
void init()
{
	ll i, j;
	cl(nex), cl(a), cl(t);
	for(i=1;i<=P;i++)a[i]=read();
	sort(a+1,a+P+1);
	for(i=1;i<=P;i++)a[P+i]=a[i]+360000;
	for(i=P*2;i;i--)a[i]-=a[i-1];
	for(i=2;i<=P;i++)t[i-1]=a[i];
	for(i=2,j=0;i<P;i++)
	{
		for(;t[j+1]!=t[i] and j;j=nex[j]);
		nex[i]=t[j+1]==t[i]?++j:0;
	}
}
void work()
{
	ll i, j, ans=0, cnt=0;
	if(P==1)
	{
		printf("%lld\n",S);
		return;
	}
	for(i=2,j=0;i<P*2;i++)
	{
		for(;t[j+1]!=a[i] and j;j=nex[j]);
		if(t[j+1]==a[i])j++;
		if(j==P-1)
		{
			ans=(ans+fastpow(S,gcd(P,i-P)))%mod;
//			printf("route %lld, ans=%lld\n",i-P,ans);
			cnt++;
			j=nex[j];
		}
	}
	ans=ans*fastpow(cnt,mod-2)%mod;
	printf("%lld\n",ans);
}
int main()
{
	while(S=read(),P=read(),S>0)
	{
		init();
		work();
	}
	return 0;
}