結果
| 問題 | No.1857 Gacha Addiction |
| ユーザー |
 ytqm3
|
| 提出日時 | 2022-02-25 22:09:19 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 465 ms / 6,000 ms |
| コード長 | 6,042 bytes |
| コンパイル時間 | 6,062 ms |
| コンパイル使用メモリ | 264,376 KB |
| 最終ジャッジ日時 | 2025-01-28 02:07:19 |
|
ジャッジサーバーID （参考情報） |
judge3 / judge2 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 43 |
ソースコード
#include<bits/stdc++.h>
#include<atcoder/all>
typedef uint64_t u64;
typedef int64_t i64;
using namespace std;
template<u64 mod> using modint=atcoder::static_modint<mod>;
template<typename T> struct comb{
vector<T> dat,idat;
comb(int mx=3000000):dat(mx+1,1),idat(mx+1,1){
for(int i=1;i<=mx;++i){
dat[i]=dat[i-1]*i;
}
idat[mx]/=dat[mx];
for(int i=mx;i>0;--i){
idat[i-1]=idat[i]*i;
}
}
T operator()(int n,int k){
if(n<0||k<0||n<k){
return 0;
}
return dat[n]*idat[k]*idat[n-k];
}
};
template<u64 mod> struct FPS{
typedef modint<mod> mint;
vector<mint> val;
FPS():val({0}){}
FPS(mint t):val({t}){}
FPS(int siz):val(max(1,siz)){}
FPS(initializer_list<modint<mod>> init):val(init){}
FPS(vector<mint> init):val(init){}
mint &operator[](int i){
return val[i];
}
int size(){
return val.size();
}
void resize(int siz){
val.resize(siz);
}
FPS& resize_(int siz){
auto tmp=val;
tmp.resize(siz);
return (*this)=tmp;
}
FPS operator-(){
for(mint& v:val){
v=mint(0)-v;
}
return (*this);
}
FPS& operator+=(mint rhs){
val[0]+=rhs;
return (*this);
}
FPS& operator-=(mint rhs){
val[0]-=rhs;
return (*this);
}
FPS& operator*=(mint rhs){
for(auto& v:val){
v*=rhs;
}
return (*this);
}
FPS& operator/=(mint rhs){
for(auto& v:val){
v/=rhs;
}
return (*this);
}
FPS operator+(mint rhs){
return FPS(*this)+=rhs;
}
FPS operator-(mint rhs){
return FPS(*this)-=rhs;
}
FPS operator*(mint rhs){
return FPS(*this)*=rhs;
}
FPS operator/(mint rhs){
return FPS(*this)/=rhs;
}
FPS& operator+=(FPS rhs){
resize(max(this->size(),rhs.size()));
for(int i=0;i<int(rhs.size());++i){
(*this)[i]+=rhs[i];
}
return (*this);
}
FPS& operator-=(FPS rhs){
resize(max(this->size(),rhs.size()));
for(int i=0;i<int(rhs.size());++i){
(*this)[i]-=rhs[i];
}
return (*this);
}
FPS& operator*=(FPS rhs){
val=atcoder::convolution(val,rhs.val);
return (*this);
}
FPS& operator/=(FPS rhs){
return (*this)*=rhs.inv();
}
FPS& operator>>=(int k){
if(int(val.size())<=k){
return (*this)={0};
}
FPS res=val;
res.val.erase(res.val.begin(),res.val.begin()+k);
return (*this)=res;
}
FPS& operator<<=(int k){
FPS res=val;
res.val.insert(res.val.begin(),k,mint(0));
return (*this)=res;
}
FPS operator+(FPS rhs){
return FPS(*this)+=rhs;
}
FPS operator-(FPS rhs){
return FPS(*this)-=rhs;
}
FPS operator*(FPS rhs){
return FPS(*this)*=rhs;
}
FPS operator/(FPS rhs){
return FPS(*this)/=rhs;
}
FPS operator<<(int k){
return FPS(*this)<<=k;
}
FPS operator>>(int k){
return FPS(*this)>>=k;
}
FPS shrink(){
for(int i=val.size()-1;i>0;--i){
if(val[i]==0){
val.pop_back();
}
else{
break;
}
}
return (*this);
}
FPS diff_(){
if(val.size()==1){
return (*this)={0};
}
FPS f(val.size()-1);
for(size_t i=1;i<val.size();++i){
f[i-1]=val[i]*i;
}
return f;
}
FPS integral_(){
FPS f(val.size()+1);
for(size_t i=0;i<val.size();++i){
f[i+1]=val[i]/(i+1);
}
return f;
}
FPS inv_(int mx=-1){
if(mx==-1){
mx=val.size();
}
if(val[0]==0){
assert(0);
}
FPS g({mint(1)/val[0]});
int now=1;
while(now<mx){
now<<=1;
FPS t=(*this);
t.resize(now);
t*=g;
t=-t+mint(2);
g*=t;
g.resize(now);
}
g.resize(mx);
return g;
}
FPS exp_(int mx=-1){
if(mx==-1){
mx=val.size();
}
if(val[0]!=0){
assert(0);
}
FPS g(mint(1));
int now=1;
while(now<mx){
now<<=1;
FPS t=(*this);
t.resize(now);
g*=t-g.log_(now)+mint(1);
g.resize(now);
}
g.resize(mx);
return g;
}
FPS log_(int mx=-1){
if(mx==-1){
mx=val.size();
}
if(val[0]!=1){
assert(0);
}
auto f=(*this);
f.resize(mx);
return (f.diff_()/f).integral_().resize_(mx);
}
FPS pow_(i64 k,int mx=-1){
if(mx==-1){
mx=val.size();
}
i64 t=0;
for(auto v:val){
if(v==0){
t++;
}
else{
break;
}
}
auto f=(*this)>>t;
if(f[0]==0){
f={0},f.resize(mx);
return f;
}
mint c=f[0];
f/=c;
(f.log(mx)*=mint(k)).exp(mx)*=c.pow(k);
if(t*k<=mx){
f<<=t*k;
f.resize(mx);
return f;
}
else{
f={0},f.resize(mx);
return f;
}
}
FPS& diff(){
return (*this)=diff_();
}
FPS& integral(){
return (*this)=integral_();
}
FPS& inv(int mx=-1){
return (*this)=inv_(mx);
}
FPS& exp(int mx=-1){
return (*this)=exp_(mx);
}
FPS& log(int mx=-1){
return (*this)=log_(mx);
}
FPS& pow(i64 k,int mx=-1){
return (*this)=pow_(k,mx);
}
void taylor_shift(mint c){
comb<mint> C(val.size());
vector<mint> g(val.size());
mint now=1;
for(size_t i=0;i<val.size();++i){
val[i]*=C.dat[i];
g[i]=now*C.idat[i];
now*=c;
}
reverse(val.begin(),val.end());
g=atcoder::convolution(val,g);
g.resize(val.size());
reverse(g.begin(),g.end());
for(size_t i=0;i<val.size();++i){
g[i]*=C.idat[i];
}
val=g;
}
};
template<u64 mod> FPS<mod> product(vector<FPS<mod>> a){
int siz=1;
while(siz<int(a.size())){
siz<<=1;
}
vector<FPS<mod>> res(siz*2-1,{1});
for(int i=0;i<int(a.size());++i){
res[i+siz-1]=a[i];
}
for(int i=siz-2;i>=0;--i){
res[i]=res[2*i+1]*res[2*i+2];
}
return res[0];
}
int main(){
typedef modint<998244353> mint;
constexpr u64 mod=998244353;
int N,S;
cin>>N>>S;
vector<FPS<mod>> f(N);
comb<mint> C;
mint invS=mint(1)/S;
for(int i=0;i<N;++i){
int p;
cin>>p;
f[i]={1,invS*p};
}
auto g=product(f);
mint ans=0;
for(int i=0;i<=N;++i){
ans+=C.dat[i]*g[i];
}
cout<<ans.val()<<endl;
}