【UOJ450】【集训队作业2018】复读机（生成函数）（单位根反演）

传送门

---

题解：

很显然一个复读机的$EGF$就是$$\sum _{i=0}^{\infty }\frac{[d\mid i]}{i!}{x}^i$$

发现$d=1,2,3$的时候模意义下都有单位根，考虑单位根反演，得到一个复读机生成函数为：

$$\frac{1}{d}\sum _{i=0}^{\infty }\frac{x^i}{i!}\sum _{j=0}^{d-1}{\omega }_d^{ij}=\frac{1}{d}\sum _{j=0}^{d-1}{e}^{{\omega }_d^{j}x}$$

$d=1$的时候直接输出$k^n$

$d=2$的时候，$k$个复读机的生成函数为$(\frac{e^x+e^{-x}}{2}{)}^k$，暴力二项式展开求第$n$项系数即可。

$d=3$的时候，$k$个复读机的生成函数为$(\frac{e^x+e^{r_{1}x}+e^{r_{2}x}}{3}{)}^k$，发现$k$有点小，还是直接暴力二项式展开就行了。

复杂度：$O(k^{d-1}\log n)$

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

cs int mod=19491001;
cs int r1=18827933,r2=663067;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline void Inc(int &a,int b){(a+=b)>=mod&&(a-=mod);}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}

int n,k,d,ans,tmp;

int fac[500005],ifac[500005];
inline int C(int n,int m){return mul(fac[n],mul(ifac[m],ifac[n-m]));}

inline void init(){
	fac[0]=1;
	for(int re i=1;i<=k;++i)fac[i]=mul(fac[i-1],i);
	ifac[k]=power(fac[k],mod-2);
	for(int i=k-1;~i;i--) ifac[i]=mul(ifac[i+1],i+1);
}

signed main(){
	std::cin>>n>>k>>d;
	switch(d){
		case 1:std::cout<<power(k,n)<<"\n";return 0;
		case 2:
			init();
			for(int re i=0;i<=k;++i)Inc(ans,mul(C(k,i),power(dec(i*2,k),n)));
			ans=mul(ans,power(power(2,mod-2),k));
			std::cout<<ans<<"\n";
			return 0;
		case 3:
			init();
			for(int re i=0;i<=k;++i){
				tmp=0;
				for(int re j=0;j<=k-i;++j)
				Inc(tmp,mul(C(k-i,j),power(add(add(mul(r2,i),mul(r1,j)),k-i-j),n)));
				Inc(ans,mul(tmp,C(k,i)));
			}
			ans=mul(ans,power(power(3,mod-2),k));
			std::cout<<ans<<"\n";
			return 0;
	}
	return 0;
}