結果
問題 | 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_popcountusing 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;}