表示这种\(Burnside\)定理之类的东西一用就忘qwq
题目要求不同染色方案数,因为变换方式只有旋转,所以只有\(n\)个置换,然后可能会出现某种方案有循环节,这个循环节长度显然要是\(\gcd(n,m)\)的因数,我们枚举循环节个数,直接套个polya然后可以得到答案为\(\frac{\sum_{d|\gcd(n,m)}f(\frac{n}{d},\frac{m}{d})\varphi(d)}{n}\),其中\(f(n,m)\)为有\(n\)个珠子,要把\(m\)个染色,并且满足最长连续的黑色个数以及首尾黑色个数\(\le k\)的方案
这个东西,可以考虑把白色的珠子提出来,然后枚举首尾放了多少个黑色\(i\),这样有\(i+1\)种方案,剩下的黑色就要放在\(n-m-1\)个空位中,并且满足一个空位不超过\(k\)个黑色.这个可以容斥,枚举一些位置已经超过\(k\)个,然后剩下随便放.这个式子长这样:$$\sum_{i=0}{k}(i+1)\sum_{j=0}{n-m-1}(-1)^j\binom{n-m-1}{j}g(m-i-j*(k+1),n-m-1)$$
其中\(g(n,m)\)表示\(n\)个一样的物品放进\(m\)个不一样的盒子的方案数,也就是\(\binom{n+m-1}{m-1}\)
代码巨丑
#include<bits/stdc++.h>
#define LL long long
#define db long double
#define il inline
#define re register
using namespace std;
const int N=1e5+10,mod=998244353;
il LL rd()
{
LL x=0,w=1;char ch=0;
while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
return x*w;
}
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int fac[N],iac[N];
int C(int n,int m){return m<0||n<m?0:1ll*fac[n]*iac[m]%mod*iac[n-m]%mod;}
int fpow(int a,int b){int an=1;while(b){if(b&1) an=1ll*an*a%mod;a=1ll*a*a%mod,b>>=1;} return an;}
int n,m,nn,kk,prm[20],nm[20],tt,ans;
int wk(int n,int m)
{
if(m<=kk) return C(n,m);
int an=0,l=n-m-1;
for(int i=0;i<=kk;++i)
{
int nw=0;
for(int j=0;j<=l&&m-i-j*(kk+1)>=0;++j)
{
int x=1ll*C(l,j)*C(m-i-j*(kk+1)+l-1,l-1)%mod;
nw=(nw+((j&1)?mod-x:x))%mod;
}
an=(an+1ll*nw*(i+1)%mod)%mod;
}
return an;
}
void dfs(int o,int d,int phi)
{
if(o>tt)
{
ans=(ans+1ll*wk(n/d,m/d)*phi%mod)%mod;
return;
}
dfs(o+1,d,phi);
for(int i=1;i<=nm[o];++i)
{
d*=prm[o];
dfs(o+1,d,phi*(prm[o]-1)*fpow(prm[o],i-1));
}
}
int main()
{
fac[0]=1;
for(int i=1;i<=N-10;++i) fac[i]=1ll*fac[i-1]*i%mod;
iac[N-10]=fpow(fac[N-10],mod-2);
for(int i=N-10;i;--i) iac[i-1]=1ll*iac[i]*i%mod;
n=rd(),m=rd(),kk=rd();
if(n==m) return printf("%d\n",(bool)!kk),0;
nn=gcd(n,m);
int x=nn,sqt=sqrt(nn);
tt=0;
for(int i=2;i<=sqt&&x>1;++i)
if(x%i==0)
{
prm[++tt]=i,nm[tt]=0;
while(x%i==0) ++nm[tt],x/=i;
}
if(x>1) prm[++tt]=x,nm[tt]=1;
ans=0,dfs(1,1,1);
printf("%lld\n",1ll*ans*fpow(n,mod-2)%mod);
return 0;
}