結果
問題 | No.2867 NOT FOUND 404 Again |
ユーザー | Tatsu_mr |
提出日時 | 2024-08-30 22:41:07 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 543 ms / 3,000 ms |
コード長 | 1,303 bytes |
コンパイル時間 | 4,491 ms |
コンパイル使用メモリ | 269,600 KB |
実行使用メモリ | 152,800 KB |
最終ジャッジ日時 | 2024-08-30 22:41:22 |
合計ジャッジ時間 | 14,114 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = atcoder::modint998244353; int main() { string s; cin >> s; int n = s.size(); string t = "404"; vector<vector<vector<mint>>> dp(n + 1, vector<vector<mint>>(2, vector<mint>(4))); dp[0][0][0] = mint(1); for (int i = 0; i < n; i++) { for (int j = 0; j < 2; j++) { for (int k = 0; k < 4; k++) { for (int l = 0; l < 10; l++) { if (j == 0 && s[i] - '0' < l) { continue; } int ni = i + 1; int nj = j | (l < s[i] - '0'); int nk = k; if (k < 3) { if (l == t[k] - '0') { nk = k + 1; } else if (l == t[0] - '0') { nk = 1; } else { nk = 0; } } dp[ni][nj][nk] += dp[i][j][k]; } } } } mint ans = 0; for (int j = 0; j < 2; j++) { for (int k = 0; k < 3; k++) { ans += dp[n][j][k]; } } ans--; cout << ans.val() << endl; }