結果

問題 No.1533 Don't be Negative!
コンテスト
ユーザー googol_S0
提出日時 2021-05-17 16:32:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,326 bytes
コンパイル時間 5,006 ms
コンパイル使用メモリ 265,660 KB
最終ジャッジ日時 2025-01-21 13:25:23
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other WA * 32 RE * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
using mint=modint998244353;
int main(){
  int N,M;
  cin>>N>>M;
  vector<vector<mint>> X(15,vector<mint>(0));
  vector<vector<vector<mint>>> Y(15,vector<vector<mint>>(0));
  X[0].assign(2*M+1,1);
  for(int i=0;i<14;i++){
    X[i+1]=convolution(X[i],X[i]);
  }for(int i=0;i<15;i++){
    Y[i].assign(1<<(15-i),vector<mint>(1,1));
  }
  Y[14][1]=X[14];
  vector<int> t(17,0);
  t[1]=1;
  for(int i=2;i<17;i++){
    t[i]=t[i-1]*2;
  }
  for(int i=13;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+2]-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;
  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][M])*V*(N-i+1);
  }
  cout<<(ANS/2).val()<<"\n";
}
0