bzoj4314 倍数？倍数！加强版

问题：求$[1,n]$中选$m$个互不相同的数使得他们的和膜$n$等于$k$的方案数

$\text{Joker}$太神了！

题解：如果没有$m$的限制，实际上相当于求${\prod }_{i=1}^n(1+x^i)$所有次数是膜$n$为$k$的项的系数和
考虑用单位根来计算这个

可以发现，设$F(x)={\prod }_{i=1}^n(1+x^i)$，则答案为

$$ans=\frac{1}{n}\sum _{i=1}^n{w}_n^{-ik}F(w_n^i)=\frac{1}{n}\sum _{i=1}^n{w}_n^{-ik}\prod _{j=1}^n(1+w_n^{ij})$$

令$d=gcd(i,n)$，则$w_n^{ij}=w_{n/d}^{i/d\ast j}$，即

$$\begin{gathered} ans=\frac{1}{n}\sum _{i=1}^n{w}_n^{-ik}\prod _{j=1}^n(1+w_{n/d}^{i/d\ast j}) \\ =\frac{1}{n}\sum _{d|n}(\sum _{j=1}^{n/d}{w}_n^{-jdk}[(j,n/d)=1])[\prod _{j=1}^{n/d}(1+w_{n/d}^j){]}^d \end{gathered}$$
考虑单位根的定义，有

$(x^n-1)=\prod (x-w_n^i)$

把$x=1$带入，并将$w_n^i$符号取反可得

$\prod (1+w_n^i)=1-(-1{)}^n$

即：
$$ans=\frac{1}{n}\sum _{d|n}(\sum _{j=1}^{n/d}{w}_n^{-jdk}[(j,n/d)=1])[1-(-1{)}^{n/d}{]}^d$$

令$f(d)={\sum }_{j=1}^{n/d}{w}_n^{-jdk}[(j,n/d)=1]$，考虑如何算这个，可以考虑容斥，令$g(d)={\sum }_{j=1}^{n/d}{w}_n^{-jdk}={\sum }_{j=1}^{n/d}{w}_{n/d}^{-jk}$，则$f(d)=g(d)-{\sum }_{d|x,x̸=d}f(x)$

考虑$g(d)$，显然当$n/d|k$时$g(d)=n/d$，否则$g(d)=0$，可以闭上眼睛感受一下，就是这些单位根转了一圈刚好抵消的样子。。。

考虑$m$的限制，其实直接改成$F(x)=\prod (1+x^{i}y)$即可，最后要求$y^m$。所以最后的式子就这样子：

$$ans=\frac{1}{n}\sum _{d|n}f(d)[1-(-y{)}^{n/d}{]}^d$$

枚举$n$的约数计算这个即可。