結果

問題 No.1533 Don't be Negative!
ユーザー googol_S0googol_S0
提出日時 2021-05-23 18:13:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 784 ms / 8,000 ms
コード長 1,484 bytes
コンパイル時間 5,836 ms
コンパイル使用メモリ 277,012 KB
実行使用メモリ 93,836 KB
最終ジャッジ日時 2024-10-11 22:34:22
合計ジャッジ時間 42,156 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 181 ms
29,828 KB
testcase_01 AC 702 ms
81,384 KB
testcase_02 AC 770 ms
93,708 KB
testcase_03 AC 774 ms
93,712 KB
testcase_04 AC 369 ms
51,592 KB
testcase_05 AC 701 ms
81,508 KB
testcase_06 AC 704 ms
81,380 KB
testcase_07 AC 372 ms
51,464 KB
testcase_08 AC 369 ms
51,460 KB
testcase_09 AC 771 ms
93,704 KB
testcase_10 AC 702 ms
81,384 KB
testcase_11 AC 778 ms
93,708 KB
testcase_12 AC 370 ms
51,384 KB
testcase_13 AC 181 ms
29,828 KB
testcase_14 AC 182 ms
29,828 KB
testcase_15 AC 372 ms
51,592 KB
testcase_16 AC 704 ms
81,388 KB
testcase_17 AC 183 ms
29,832 KB
testcase_18 AC 702 ms
81,388 KB
testcase_19 AC 701 ms
81,384 KB
testcase_20 AC 371 ms
51,464 KB
testcase_21 AC 703 ms
81,380 KB
testcase_22 AC 371 ms
51,464 KB
testcase_23 AC 183 ms
29,956 KB
testcase_24 AC 372 ms
51,588 KB
testcase_25 AC 776 ms
93,836 KB
testcase_26 AC 779 ms
93,724 KB
testcase_27 AC 370 ms
51,588 KB
testcase_28 AC 709 ms
81,380 KB
testcase_29 AC 706 ms
81,384 KB
testcase_30 AC 779 ms
93,836 KB
testcase_31 AC 702 ms
81,388 KB
testcase_32 AC 705 ms
81,436 KB
testcase_33 AC 779 ms
93,708 KB
testcase_34 AC 373 ms
51,592 KB
testcase_35 AC 184 ms
29,956 KB
testcase_36 AC 778 ms
93,624 KB
testcase_37 AC 774 ms
93,704 KB
testcase_38 AC 778 ms
93,712 KB
testcase_39 AC 707 ms
81,384 KB
testcase_40 AC 707 ms
81,388 KB
testcase_41 AC 706 ms
81,384 KB
testcase_42 AC 780 ms
93,708 KB
testcase_43 AC 706 ms
81,508 KB
testcase_44 AC 707 ms
81,384 KB
testcase_45 AC 708 ms
81,384 KB
testcase_46 AC 706 ms
81,352 KB
testcase_47 AC 776 ms
93,712 KB
testcase_48 AC 706 ms
81,384 KB
testcase_49 AC 705 ms
81,384 KB
testcase_50 AC 779 ms
93,704 KB
testcase_51 AC 780 ms
93,704 KB
testcase_52 AC 784 ms
93,704 KB
testcase_53 AC 773 ms
93,832 KB
testcase_54 AC 772 ms
93,708 KB
testcase_55 AC 776 ms
93,836 KB
testcase_56 AC 772 ms
93,696 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
using mint=modint998244353;
const int log2NMAX=15;
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";
}
0