結果
問題 | No.1387 Mitarushi's Remodeling |
ユーザー |
![]() |
提出日時 | 2021-02-10 01:01:50 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 567 ms / 2,000 ms |
コード長 | 1,565 bytes |
コンパイル時間 | 3,694 ms |
コンパイル使用メモリ | 189,756 KB |
最終ジャッジ日時 | 2025-01-18 16:51:18 |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 67 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #include <list> #include <atcoder/all> #define popcount __builtin_popcount using namespace std; using namespace atcoder; typedef long long ll; typedef pair<int, int> P; using mint=modint998244353; mint f[2000010], invf[2000010]; void fac(int n){ f[0]=1; for(ll i=1; i<=n; i++) f[i]=f[i-1]*i; invf[n]=f[n].inv(); for(ll i=n-1; i>=0; i--) invf[i]=invf[i+1]*(i+1); } mint comb(int x, int y){ if(!(0<=y && y<=x)) return 0; return f[x]*invf[y]*invf[x-y]; } mint fs[500050]; int main() { int n, k; cin>>n>>k; vector<mint> s(n), sr(n); mint ss=0, ss2=0; mint mp=1; for(int i=0; i<n; i++){ ll m; cin>>m; mp*=mint(m); s[i]=mint(m+1)/2; sr[n-1-i]=s[i]; ss+=s[i]; ss2+=s[i]*s[i]; } fac(n); mint ps=0; for(int i=1; i<=k; i++){ ps+=f[n-1]*invf[n-1-i]; } mint ans=(ss*ss-ss2)/2*ps; auto sp=convolution(s, sr); for(int i=0; i<=n; i++){ fs[i+1]=fs[i]+invf[i]; } for(int d=0; d<n-1; d++){ ans-=f[d]*(fs[d]-fs[max(0, d-k)])*sp[d]; } ans*=mp; cout<<ans.val()<<endl; return 0; }