結果
問題 |
No.1533 Don't be Negative!
|
ユーザー |
![]() |
提出日時 | 2021-05-23 18:12:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,484 bytes |
コンパイル時間 | 5,098 ms |
コンパイル使用メモリ | 264,992 KB |
最終ジャッジ日時 | 2025-01-21 17:48:41 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 32 RE * 21 |
ソースコード
#include<bits/stdc++.h> #include<atcoder/all> using namespace std; using namespace atcoder; typedef long long ll; using mint=modint998244353; const int log2NMAX=14; int main(){ int N,M,K; cin>>N>>M>>K; vector<vector<mint>> X(log2NMAX+1,vector<mint>(0)); vector<vector<vector<mint>>> Y(log2NMAX+1,vector<vector<mint>>(0)); X[0].assign(2*M+1,1); X[0][M-K]=0; X[0][M+K]=0; for(int i=0;i<log2NMAX;i++){ X[i+1]=convolution(X[i],X[i]); }for(int i=0;i<log2NMAX+1;i++){ Y[i].assign(1<<(log2NMAX+1-i),vector<mint>(1,1)); } Y[log2NMAX][1]=X[log2NMAX]; vector<int> t(log2NMAX+3,0); t[1]=1; for(int i=2;i<log2NMAX+3;i++){ t[i]=t[i-1]*2; } for(int i=log2NMAX-1;i>=0;i--){ for(int j=0;j<(int)(Y[i+1].size());j++){ Y[i][j<<1]=Y[i+1][j]; Y[i][(j<<1)+1]=convolution(Y[i+1][j],X[i]); vector<mint> Z((t[i+1]-1)*2*M+1); if(Y[i][j<<1].size()>=Z.size()){ for(int k=0;k<(int)(Z.size());k++){ Z[k]=Y[i][j<<1][(Y[i][j<<1].size()>>1)+k-(Z.size()>>1)]; } Y[i][j<<1]=Z; }if(Y[i][(j<<1)+1].size()>=Z.size()){ for(int k=0;k<(int)(Z.size());k++){ Z[k]=Y[i][(j<<1)+1][(Y[i][(j<<1)+1].size()>>1)+k-(Z.size()>>1)]; } Y[i][(j<<1)+1]=Z; } } } mint ANS=0; mint P=2*M-1; if(K==0){ P+=1; } mint V=P.pow(N); mint W=P.inv(); mint U=1; for(int i=1;i<=N;i++){ V*=W; U*=P; ANS+=(U-Y[0][i][0])*V*(N-i+1); } cout<<(ANS/2).val()<<"\n"; }