LOJ#6485. LJJ 学二项式定理（单位根反演+二项式展开）

最新推荐文章于 2023-11-21 12:28:31 发布

SC.ldxcaicai

最新推荐文章于 2023-11-21 12:28:31 发布

阅读量229

点赞数 1

分类专栏： # 单位根反演

本文链接：https://blog.csdn.net/dreaming__ldx/article/details/99479218

版权

单位根反演专栏收录该内容

2 篇文章 0 订阅

订阅专栏

传送门
~~$l d x$ 又来水水题啦~~

$\begin{aligned}Ans=&\sum\limits_{i=0}^nC_n^iS^ia_{i\%4}\\=&\sum\limits_{j=0}^3a_j\sum\limits_{i=0}^nC_n^iS^i\frac{\sum\limits_{k=0}^3(\omega_4^{i-j})^k}4\\=&\frac14\sum\limits_{j=0}^3a_j\sum\limits_{k=0}^3\omega_4^{-kj}\sum_{i=0}^nC_n^i(S\omega_4^k)^i\\=&\frac14\sum\limits_{j=0}^3a_j\sum\limits_{k=0}^3\omega_4^{-kj}(S\omega_4^k)^n\end{aligned}$
然后就做完啦。
代码：

#include<bits/stdc++.h>
#define ri register int
using namespace std;
const int rlen=1<<18|1;
inline char gc(){
	static char buf[rlen],*ib,*ob;
	(ib==ob)&&(ob=(ib=buf)+fread(buf,1,rlen,stdin));
	return ib==ob?-1:*ib++;
}
inline int read(){
	int ans=0;
	char ch=gc();
	while(!isdigit(ch))ch=gc();
	while(isdigit(ch))ans=((ans<<2)+ans<<1)+(ch^48),ch=gc();
	return ans;
}
typedef long long ll;
inline ll readl(){
	ll ans=0;
	char ch=gc();
	while(!isdigit(ch))ch=gc();
	while(isdigit(ch))ans=((ans<<2)+ans<<1)+(ch^48),ch=gc();
	return ans;
}
const int mod=998244353;
int w[4]={1,911660635,998244352,86583718};
inline int add(int a,int b){return (a+=b)<mod?a:a-mod;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){return (ll)a*b%mod;}
inline void Add(int&a,int b){(a+=b)<mod?a:(a-=mod);}
inline void Dec(int&a,int b){(a-=b)<0?(a+=mod):a;}
inline void Mul(int&a,int b){a=(ll)a*b%mod;}
inline int ksm(int a,int p){int ret=1;for(;p;p>>=1,Mul(a,a))if(p&1)Mul(ret,a);return ret;}
int a[4],s,ans,C[105][105];
ll n;
inline void init(){
	for(ri i=0;i<=100;++i){
		C[i][0]=C[i][i]=1;
		for(ri j=1;j<i;++j)C[i][j]=add(C[i-1][j],C[i-1][j-1]);
	}
}
int main(){
	#ifdef ldxcaicai
	freopen("lx.in","r",stdin);
	#endif
	init();
	for(ri t,inv=ksm(4,mod-2),ss,tt=read();tt;--tt){
		ans=0;
		n=readl()%(mod-1),s=read();
		for(ri i=0;i<4;++i)a[i]=read();
		for(ri i=0;i<4;++i){
			t=ksm(add(1,mul(s,w[i])),n);
			ss=0;
			for(ri j=0;j<4;++j)Add(ss,mul(a[j],w[(4-i*j%4)%4]));
			Add(ans,mul(t,ss));
		}
		cout<<mul(inv,ans)<<'\n';
	}
	return 0;
}