結果
問題 | No.2899 Taffy Permutation |
ユーザー | t98slider |
提出日時 | 2024-09-20 22:46:28 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 2,191 bytes |
コンパイル時間 | 4,798 ms |
コンパイル使用メモリ | 267,924 KB |
実行使用メモリ | 13,756 KB |
最終ジャッジ日時 | 2024-09-20 22:46:37 |
合計ジャッジ時間 | 8,465 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
13,756 KB |
testcase_01 | AC | 2 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,940 KB |
testcase_03 | AC | 2 ms
6,940 KB |
testcase_04 | AC | 2 ms
6,944 KB |
testcase_05 | AC | 2 ms
6,940 KB |
testcase_06 | AC | 2 ms
6,940 KB |
testcase_07 | AC | 2 ms
6,940 KB |
testcase_08 | AC | 2 ms
6,940 KB |
testcase_09 | TLE | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = atcoder::modint998244353; using ll = long long; //enumeration<mint> enu(200000);のように宣言する template<class T> struct enumeration{ int N; vector<T> fact, inv; enumeration() : N(0), fact(1, 1), inv(1, 1) {} enumeration(int _n) : N(_n), fact(_n + 1), inv(_n + 1) { fact[0] = 1; for(int i = 1; i <= N; i++) fact[i] = fact[i - 1] * i; inv[N] = T(1) / fact[N]; for(int i = N; i >= 1; i--) inv[i - 1] = inv[i] * i; } void expand(int lim){ fact.resize(lim + 1); inv.resize(lim + 1); for(int i = N + 1; i <= lim; i++) fact[i] = i * fact[i - 1]; inv[lim] = T(1) / fact[lim]; for(int i = lim; i >= N + 2; i--) inv[i - 1] = i * inv[i]; N = lim; } T Per(int n, int k){ if(k > n) return 0; if(n > N) expand(n); return fact[n] * inv[n - k]; } T C(int n, int k){ if(n < 0 || k < 0 || k > n) return 0; if(n > N) expand(n); return fact[n] * inv[n - k] * inv[k]; } T H(int n, int k){ if(n ==0 && k == 0) return 1; if(n <= 0 || k < 0) return 0; return C(n + k - 1, k); } }; int main(){ ios::sync_with_stdio(false); cin.tie(0); enumeration<mint> enu(2005); int n; string s; cin >> n >> s; int cnt = 0; reverse(s.begin(), s.end()); while(!s.empty() && s.back() == '1') cnt++, s.pop_back(); reverse(s.begin(), s.end()); const int m = s.size(); vector<mint> dp(1); dp[0] = 1; for(int i = 1; i < m; i++){ vector<mint> ndp(i + 1); if(s[i] == '0'){ for(int j = 0; j < i; j++){ for(int k = 0; k <= i; k++){ ndp[max(k, j + 1)] += dp[j]; } } }else{ for(int j = 0; j < i; j++){ for(int k = j + 1; k <= i; k++){ ndp[j] += dp[j]; } } } swap(dp, ndp); } mint ans = enu.Per(n, cnt); ans *= accumulate(dp.begin(), dp.end(), mint(0)); cout << ans.val() << '\n'; }