loj 6485 LJJ学二项式定理 —— 单位根反演

题目：https://loj.ac/problem/6485

先把 \( a_{i mod 4} \) 处理掉，其实就是

\( \sum\limits_{i=0}^{3} a_{i} \sum\limits_{j=0}^{n} C_{n}^{j} * s^{j} * [4|(j-i)] \)

然后把 \( [4|(j-i)] \) 单位根反演，得到

\( \sum\limits_{i=0}^{3} a_{i} \sum\limits_{j=0}^{n} C_{n}^{j} * s^{j} * \frac{1}{4} \sum\limits_{k=0}^{3} w_{4}^{k(j-i)} \)

\( \frac{1}{4} \sum\limits_{i=0}^{3} a_{i} \sum\limits_{k=0}^{3} \sum\limits_{j=0}^{n} C_{n}^{j} * s^{j} * w_{4}^{k(j-i)} \)

这里 \( w_{4}^{0} = g^{\frac{mod-1}{4}} \)，\( g \) 是 998244353 的原根；

然后为了有二项式定理的形式，构造 \( F(i,w_{4}^{-k}) = w_{4}^{-k(i-n)} ( w_{4}^{-k} + s)^{n} \)

于是 \( ans = \frac{1}{4} \sum\limits_{i=0}^{3} a_{i} \sum\limits_{k=0}^{3} F(i,w_{4}^{-k}) \)

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const mod=,g=;
ll rd()
{
  ll ret=,f=; char ch=getchar();
  while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return f?ret:-ret;
}
ll pw(ll a,ll b)
{
  a=(a%mod+mod)%mod; b=(b%(mod-)+(mod-))%(mod-);
  ll ret=;
  for(;b;b>>=,a=a*a%mod)if(b&)ret=ret*a%mod; return ret;
}
int upt(int x){while(x>=mod)x-=mod; while(x<)x+=mod; return x;}
int s,a[]; ll n;
int F(int i,int x)
{
  int ret=upt(x+s); ret=pw(ret,n);
  return (ll)ret*pw(x,i-n)%mod;
}
int main()
{
  int T=rd();
  while(T--)
    {
      n=rd(); s=rd(); for(int i=;i<;i++)a[i]=rd();
      int w=pw(g,(mod-)/),inv=pw(w,mod-);
      int ans=;
      for(int i=;i<=;i++)
    for(int k=,t=;k<=;k++,t=(ll)t*inv%mod)
      ans=(ans+(ll)F(i,t)*a[i])%mod;
      ans=(ll)ans*pw(,mod-)%mod;
      printf("%d\n",upt(ans));
    }
  return ;
}