結果
問題 | No.2899 Taffy Permutation |
ユーザー | karinohito |
提出日時 | 2024-09-20 23:32:59 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 1,537 bytes |
コンパイル時間 | 4,892 ms |
コンパイル使用メモリ | 267,356 KB |
実行使用メモリ | 34,816 KB |
最終ジャッジ日時 | 2024-09-20 23:33:05 |
合計ジャッジ時間 | 6,180 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 2 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 2 ms
5,376 KB |
testcase_05 | AC | 2 ms
5,376 KB |
testcase_06 | AC | 2 ms
5,376 KB |
testcase_07 | AC | 2 ms
5,376 KB |
testcase_08 | AC | 2 ms
5,376 KB |
testcase_09 | AC | 42 ms
34,816 KB |
testcase_10 | AC | 42 ms
34,816 KB |
testcase_11 | AC | 42 ms
34,688 KB |
testcase_12 | AC | 42 ms
34,560 KB |
testcase_13 | AC | 41 ms
34,688 KB |
testcase_14 | AC | 42 ms
34,560 KB |
testcase_15 | AC | 42 ms
34,560 KB |
testcase_16 | AC | 42 ms
34,432 KB |
testcase_17 | AC | 42 ms
34,560 KB |
testcase_18 | AC | 2 ms
5,376 KB |
testcase_19 | AC | 43 ms
34,432 KB |
ソースコード
#include<bits/stdc++.h> using namespace std; using ll=long long; #include<atcoder/all> using namespace atcoder; using mint = modint998244353; vector<mint> fact, factinv, inv; const ll mod = 998244353; void prenCkModp(ll n) { fact.resize(n + 5); factinv.resize(n + 5); inv.resize(n + 5); fact[0] = fact[1] = 1; factinv[0] = factinv[1] = 1; inv[1] = 1; for (ll i = 2; i < n + 5; i++) { fact[i] = (fact[i - 1] * i); inv[i] = mod - (inv[mod % i] * (mod / i)); factinv[i] = (factinv[i - 1] * inv[i]); } } mint nCk(ll n, ll k) { if (n < k||k<0) return 0; return fact[n] * (factinv[k] * factinv[n - k]); } int main(){ cin.tie(nullptr); ios::sync_with_stdio(false); ll N; cin>>N; string S; cin>>S; vector<vector<mint>> DP(N+1,vector<mint>(N+1,0)); vector<vector<mint>> SDP(N+1,vector<mint>(N+1,0)); DP[0][0]=1; for(int i=0;i<=N;i++)SDP[0][i]=1; for(int i=0;i<N;i++){ if(S[i]=='1'){ for(int j=0;j<=i;j++){ ll d=i-j; DP[i+1][j]+=DP[i][j]*(d+1); } } else{ for(int j=1;j<=i+1;j++){ DP[i+1][j]+=SDP[i][j-1]; if(j!=1){ DP[i+1][j]+=DP[i][j-1]*(j-1); } } } SDP[i+1][0]=DP[i+1][0]; for(int j=1;j<=N;j++){ SDP[i+1][j]=DP[i+1][j]+SDP[i+1][j-1]; } } // cout<<SDP[1][1].val()<<endl; cout<<SDP[N][N].val()<<endl; }