結果

問題 No.2867 NOT FOUND 404 Again
コンテスト
ユーザー vjudge1
提出日時 2024-09-12 11:22:29
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 673 ms / 3,000 ms
コード長 1,431 bytes
コンパイル時間 1,791 ms
コンパイル使用メモリ 170,948 KB
実行使用メモリ 113,748 KB
最終ジャッジ日時 2024-09-12 11:22:43
合計ジャッジ時間 12,909 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 5, mod = 998244353;

int n, dp[N][3][3][3], x[N];

string s;

int val(int e) {
  if (e == 0) {
    return 0;
  }
  else if (e == 4) {
    return 1;
  }
  return 2;
}

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);
  cin >> s;
  n = s.size();
  s = " " + s;
  for (int i = 1; i <= n; i++) {
    x[i] = (int)(s[i] - '0');
  }
  //0?? = 0,1 ?? = 4, 2???????
  dp[0][0][0][1] = 1;
  for (int i = 1; i <= n; i++) {
    for (int a = 0; a <= 2; a++) {
      for (int b = 0; b <= 2; b++) {
        for (int c = 0; c <= 9; c++) {
          if (c == 4 && b == 0 && a == 1) {
            continue;
          }
          if (c < x[i]) {
            dp[i][val(c)][b][0] = (dp[i][val(c)][b][0] + dp[i - 1][b][a][1]) % mod;
            dp[i][val(c)][b][0] = (dp[i][val(c)][b][0] + dp[i - 1][b][a][0]) % mod;
          }
          else if (c == x[i]) {
            dp[i][val(c)][b][1] = (dp[i][val(c)][b][1] + dp[i - 1][b][a][1]) % mod;
            dp[i][val(c)][b][0] = (dp[i][val(c)][b][0] + dp[i - 1][b][a][0]) % mod;
          }
          else {
            dp[i][val(c)][b][0] = (dp[i][val(c)][b][0] + dp[i - 1][b][a][0]) % mod;
          }
        }
      }
    }
  }
  int ans = 0;
  for (int a = 0; a <= 2; a++) {
    for (int b = 0; b <= 2; b++) {
      ans += (dp[n][a][b][0] + dp[n][a][b][1]) % mod;
      ans %= mod;
    }
  }
  cout << ans - 1;
  return 0;
}
0