結果

問題 No.1417 100の倍数かつ正整数(2)
ユーザー kusaf_kusaf_
提出日時 2024-10-01 17:50:58
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 3,000 ms
コード長 2,364 bytes
コンパイル時間 3,129 ms
コンパイル使用メモリ 251,876 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-10-01 17:51:03
合計ジャッジ時間 4,646 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using namespace atcoder;
using ll = long long;
using mint = modint1000000007;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  string S;
  cin >> S;
  int N = S.size();

  static mint dp[10010][3][3][2][2][2]{};
  dp[0][0][0][0][0][0] = 1;
  for(int i = 0; i < N; i++) {
    for(int j = 0; j < 10; j++) {
      for(int k = 0; k < 3; k++) {
        for(int l = 0; l < 3; l++) {
          int nk = k, nl = l + (j == 5);
          if(j == 2 || j == 6) { nk++; }
          if(j == 4) { nk += 2; }
          if(j == 8) { nk += 3; }
          nk = min(2, nk);
          nl = min(2, nl);
          if(j < S[i] - '0') {
            if(!j) {
              dp[i + 1][nk][nl][0][1][0] += dp[i][k][l][0][0][0] + dp[i][k][l][0][1][0];
              dp[i + 1][nk][nl][1][1][1] += dp[i][k][l][0][0][1] + dp[i][k][l][0][1][1] + dp[i][k][l][1][0][1] + dp[i][k][l][1][1][1];
            }
            else {
              dp[i + 1][nk][nl][0][1][1] += dp[i][k][l][0][0][0] + dp[i][k][l][0][1][0] + dp[i][k][l][0][0][1] + dp[i][k][l][0][1][1];
              dp[i + 1][nk][nl][1][1][1] += dp[i][k][l][1][0][0] + dp[i][k][l][1][1][0] + dp[i][k][l][1][0][1] + dp[i][k][l][1][1][1];
            }
          }
          else if(j == S[i] - '0') {
            if(!j) {
              dp[i + 1][nk][nl][0][0][0] += dp[i][k][l][0][0][0];
              dp[i + 1][nk][nl][0][1][0] += dp[i][k][l][0][1][0];
              dp[i + 1][nk][nl][1][0][1] += dp[i][k][l][0][0][1] + dp[i][k][l][1][0][1];
              dp[i + 1][nk][nl][1][1][1] += dp[i][k][l][0][1][1] + dp[i][k][l][1][1][1];
            }
            else {
              dp[i + 1][nk][nl][0][0][1] += dp[i][k][l][0][0][0] + dp[i][k][l][0][0][1];
              dp[i + 1][nk][nl][0][1][1] += dp[i][k][l][0][1][0] + dp[i][k][l][0][1][1];
              dp[i + 1][nk][nk][1][0][1] += dp[i][k][l][1][0][0] + dp[i][k][l][1][0][1];
              dp[i + 1][nk][nk][1][1][1] += dp[i][k][l][1][1][0] + dp[i][k][l][1][1][1];
            }
          }
          else {
            dp[i + 1][nk][nl][0][1][1] += dp[i][k][l][0][1][0] + dp[i][k][l][0][1][1];
            dp[i + 1][nk][nl][1][1][1] += dp[i][k][l][1][1][0] + dp[i][k][l][1][1][1];
          }
        }
      }
    }
  }

  cout << (dp[N][2][2][0][0][1] + dp[N][2][2][0][1][1]).val() << "\n";
}
0