【LOJ6485】LJJ 学二项式定理（单位根反演）

传送门

---

题解：

我们要求的是这个东西：
$$Ans=\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^i\cdot a_{i\%4}$$

首先我们有单位根的性质：$$[n\mid k]=\frac{1}{n}\sum _{i=0}^{n-1}{\omega }_n^{ik}$$

我们考虑求一个数列 { a } \{a\} {a}的所有$k$的倍数项之和。令其生成函数为$A(x)=\sum _{i=0}^n{a}_i{x}^i$

那么我们有$$\sum _{i=0}^n{a}_i[k\mid i]=\frac{1}{k}\sum _{i=0}^{k-1}A({\omega }_k^i)$$

如果$A({\omega }_k^i)$有什么快速计算方式的话就可以优化复杂度了。

以上就是单位根反演的理论基础。

在这道题中我们先将上面的式子进行一点转化：

$$Ans=\sum _{d=0}^3{a}_d\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^i[i\%4=d]$$

先不管那个$d$，我们考虑$d=0$的时候怎么做。

我们要求$$\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^{n-i}[4\mid i]$$

令$a_i=\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^{n-i}$，注意这里翻转了一下，则我们由二项式定理知道：$A(x)=(x+s{)}^n$

于是

$$\begin{array}{rlr}\sum _{i=0}^n\left(\genfrac{}{}{0ex}{}{n}{i}\right)s^{n-i}[4\mid i]& =& \frac{1}{4}\sum _{i=0}^3({\omega }_4^i+s{)}^n\end{array}$$

于是可以直接快速幂解决。

那么当$d$不为$0$的情况呢？

生成函数系数平移啊，也就是乘上若干个$x$。

于是乎，很愉快地，我们的答案出来了：

$$\frac{1}{4}\sum _{d=0}^3{a}_{(n+d)\%4}\sum _{i=0}^3({\omega }_4^i+s{)}^n{\omega }_4^{di}$$

快速幂+递推就行了。

---

代码：

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

cs int mod=998244353,inv4=748683265;
cs int w0=1,w1=911660635,w2=998244352,w3=86583718;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline void Inc(int &a,int b){(a+=b)>=mod&&(a-=mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
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;
}

ll n;
int nn,s,a[4],T;
int f0,f1,f2,f3,res;

signed main(){
	scanf("%d",&T);
	while(T--){
		scanf("%lld%d%d%d%d%d",&n,&s,&a[0],&a[1],&a[2],&a[3]);
		res=0,nn=n%(mod-1);
		f0=power(add(w0,s),nn);
		f1=power(add(w1,s),nn);
		f2=power(add(w2,s),nn);
		f3=power(add(w3,s),nn);
		Inc(res,mul(a[nn&3],add(add(f0,f1),add(f2,f3))));
		f1=mul(f1,w1);f2=mul(f2,w2);f3=mul(f3,w3);
		Inc(res,mul(a[nn+1&3],add(add(f0,f1),add(f2,f3))));
		f1=mul(f1,w1);f2=mul(f2,w2);f3=mul(f3,w3);
		Inc(res,mul(a[nn+2&3],add(add(f0,f1),add(f2,f3))));
		f1=mul(f1,w1);f2=mul(f2,w2);f3=mul(f3,w3);
		Inc(res,mul(a[nn+3&3],add(add(f0,f1),add(f2,f3))));
		std::cout<<mul(res,inv4)<<"\n";
	}
	return 0;
}