「HDU 2865」Birthday Toy

传送门

---

problem

$n$ 个小球组成正 $n$ 边形，其中相邻的两个小球用等长的木棍相连。正 $n$ 边形中间有一个大球，这个大球与 $n$ 个小球分别用木棍相连。若用 $k$ 种颜色对这 $n+1$ 个球染色，且一个木棍相连的两个点不能染成同一种颜色，问方案数。

两种染色方案相同，当且仅当其中一种可以经过若干次旋转变成另一种。

数据范围：$3\le n\le 10^9$，$4\le k\le 10^9$，且至少有 $1000$ 组数据。

---

solution

首先，由于有限制，不能直接用 Polya 定理，我们考虑 Burnside 引理。

显然，给大圆随便涂一个颜色，算一下用剩下的 $k-1$ 种颜色去染 $n$ 个小圆的方案数，再乘 $k$ 便是答案。

注：为了方便，以下的 $k$ 代表的是 $k-1$。

对于第 $i$ 个置换，它有 $\gcd (n,i)$ 个循环，由于每个循环的涂色必须相同，可以看成一个点，所以我们只需求对环上 $\gcd (n,i)$ 个点染色，保证相邻两个点涂色不同的方案数。

设 $f_i$ 表示由 $i$ 个点组成的环的涂色方案，那么：

$$f_i=(k-1)f_{i-2}+(k-2)f_{i-1}$$

具体就是，假设 $1$ 和 $i-1$ 颜色不同，贡献是 $(k-2)f_{i-1}$；如果颜色相同，贡献是 $(k-1)f_{i-2}$。

那么这个显然是可以用矩阵快速幂优化的东西。

然后还有个和「POJ 2154」Color 一样的优化，即用 $\phi$ 优化。

时间复杂度应该是 $O(\sqrt{n}\log n)$ 左右的。

---

code

#include<bits/stdc++.h>
using namespace std;
const int N=5e4+5,P=1e9+7;
int n,k;
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
int power(int a,int b){
	int ans=1;
	for(;b;b>>=1,a=mul(a,a))  if(b&1)  ans=mul(ans,a);
	return ans;
}
struct matrix{
	int m[2][2];
	matrix(int t=0)  {m[0][0]=m[1][1]=t,m[0][1]=m[1][0]=0;}
	friend matrix operator*(const matrix &A,const matrix &B){
		matrix C(0);
		C.m[0][0]=add(mul(A.m[0][0],B.m[0][0]),mul(A.m[0][1],B.m[1][0]));
		C.m[0][1]=add(mul(A.m[0][0],B.m[0][1]),mul(A.m[0][1],B.m[1][1]));
		C.m[1][0]=add(mul(A.m[1][0],B.m[0][0]),mul(A.m[1][1],B.m[1][0]));
		C.m[1][1]=add(mul(A.m[1][0],B.m[0][1]),mul(A.m[1][1],B.m[1][1]));
		return C;
	}
	friend matrix operator^(matrix A,int b){
		matrix ans(1);
		for(;b;b>>=1,A=A*A)  if(b&1)  ans=ans*A;
		return ans;
	}
}A,B;
void init(){
	A.m[0][0]=k-2,A.m[0][1]=k-1,A.m[1][0]=1,A.m[1][1]=0;
	B.m[0][0]=mul(k,k-1),B.m[0][1]=0;
}
int sum,prime[N],mark[N];
void linear_sieves(){
	for(int i=2;i<N;++i){
		if(!mark[i])  prime[++sum]=i;
		for(int j=1;j<=sum&&i*prime[j]<N;++j){
			mark[i*prime[j]]=1;
			if(i%prime[j]==0)  break;
		}
	}
}
int solve(int x){
	if(x==1)  return 0;
	if(x==2)  return mul(k,k-1);
	return ((A^(x-2))*B).m[0][0];
}
int phi(int n){
	int ans=n;
	for(int i=1;prime[i]*prime[i]<=n;++i){
		if(n%prime[i]==0){
			ans=ans/prime[i]*(prime[i]-1);
			while(n%prime[i]==0)  n/=prime[i];
		}
	}
	if(n!=1)  ans=ans/n*(n-1);
	return ans;
}
int main(){
	linear_sieves();
	while(~scanf("%d%d",&n,&k)){
		int ans=0;--k,init();
		for(int i=1;i*i<=n;++i){
			if(n%i==0){
				ans=add(ans,mul(solve(i),phi(n/i)));
				if(i*i!=n)  ans=add(ans,mul(solve(n/i),phi(i)));
			}
		}
		ans=mul(ans,mul(k+1,power(n,P-2)));
		printf("%d\n",ans);
	}
	return 0;
}