单位根反演&[loj6485]LJJ 学二项式定理

前言

之前写反演的博客对于单位根反演只提了FFT
这里补一下一个应用

单位根反演

$$f_i=\sum _{j=0}^{n-1}{\omega }_n^{i\ast j}{g}_j\iff g_i=\sum _{j=0}^{n-1}\frac{{\omega }_n^{-i\ast j}}{n}{f}_j$$
这是FFT里的
证明很方便，见我的博客对于容斥原理&反演的思考和总结
事实上，我们思考上式的本质
我们发现就是
$$[n|x]=\frac{1}{n}\sum _{i=0}^{n-1}({\omega }_n^x{)}^i$$
FFT中的形式把$x$取值$a-b$，又保证$|a-b|<n$那么$n|x$只有在$a=b$的时候满足，即${\delta }_{a,b}$
所以根据这个我们可以得到更多的应用，常见形式如下
对一个多项式
$$f(x)=\sum _{i=0}^{n-1}{a}_i{x}^i$$
求所有次数为$k$的倍数的系数之和
$$\begin{array}{rl}Ans& =\sum _{i=0}^{n-1}[k|i]a_i\end{array}$$

题解

题目要求的是
$$\left[\sum _{i=0}^n\left(\left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot s^i\cdot a_{imod4}\right)\right]\mathrm{m}\mathrm{o}\mathrm{d}998244353$$
我们对$i\mathrm{m}\mathrm{o}\mathrm{d}4$的值进行分别讨论，我们现在要求的就是各组的系数和
我们现在要求
$$\left[\sum _{i=0}^n\left(\left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot s^i[4|i]\right)\right]\mathrm{m}\mathrm{o}\mathrm{d}998244353$$
$$\begin{array}{rl}\sum _{i=0}^n\left(\left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot s^i[4|i]\right)& =\sum _{i=0}^n\left(\left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot s^i\frac{1}{4}\sum _{j=0}^{4-1}({\omega }_4^i{)}^j\right)\\ & =\frac{1}{4}\sum _{j=0}^3\sum _{i=0}^n\left(\left(\genfrac{}{}{0ex}{}{n}{i}\right)\cdot s^i({\omega }_4^j{)}^i\right)\\ & =\frac{1}{4}\sum _{j=0}^3{\left(s{\omega }_4^j+1\right)}^n\end{array}$$
那么单个的系数就求得了，那么我们发现除了$[4|i]$的情况，还有$[4|i-1]$,$[4|i-2]$,$[4|i-3]$的情况，那么相当于是我们将关于$x={\omega }_4^j$的函数的值分别乘以$x^{-1}，x^{-2}，x^{-3}$，然后（具体做法是，对于$x={\omega }_4^i$，$x^{-1}={\omega }_4^{-i}$），其它的一样做即可

代码

#include<bits/stdc++.h>
typedef long long ll;
#define rg register
template <typename T> inline void read(T&x){char cu=getchar();x=0;bool fla=0;while(!isdigit(cu)){if(cu=='-')fla=1;cu=getchar();}while(isdigit(cu))x=x*10+cu-'0',cu=getchar();if(fla)x=-x;}
template <typename T> inline void printe(const T x){if(x>=10)printe(x/10);putchar(x%10+'0');}
template <typename T> inline void print(const T x){if(x<0)putchar('-'),printe(-x);else printe(x);}
const ll mod=998244353;
ll T,n,s,a0,a1,a2,a3,w0,w1,w2,w3,inv;
inline ll pow(ll x,ll y)
{
	ll res=1;
	for(;y;y>>=1,x=x*x%mod)if(y&1)res=res*x%mod;
	return res;
}
int main()
{
	read(T);
	w0=1,w1=pow(3,(mod-1)/4),w2=w1*w1%mod,w3=w1*w2%mod;
	inv=pow(4,mod-2);
	while(T--)
	{
		read(n),read(s),read(a0),read(a1),read(a2),read(a3);
		const ll k0=pow(s*w0%mod+1,n),k1=pow(s*w1%mod+1,n),k2=pow(s*w2%mod+1,n),k3=pow(s*w3%mod+1,n);
		ll ans=0;
		ans=(ans+(k0*w0+k1*w0+k2*w0+k3*w0)%mod*a0)%mod;
		ans=(ans+(k0*w0+k1*w3+k2*w2+k3*w1)%mod*a1)%mod;
		ans=(ans+(k0*w0+k1*w2+k2*w0+k3*w2)%mod*a2)%mod;
		ans=(ans+(k0*w0+k1*w1+k2*w2+k3*w3)%mod*a3)%mod;
		print(ans*inv%mod),putchar('\n');
	}
	return 0;
}

总结

单位根反演需要推式子，但写起来比较清真