結果

問題 No.315 世界のなんとか3.5
ユーザー t8m8⛄️
提出日時 2017-06-23 15:11:26
言語 Nim
(2.2.0)
結果
WA  
実行時間 -
コード長 1,636 bytes
コンパイル時間 3,697 ms
コンパイル使用メモリ 71,808 KB
実行使用メモリ 817,408 KB
最終ジャッジ日時 2024-06-30 01:30:34
合計ジャッジ時間 8,585 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 12 WA * 2 MLE * 3 -- * 19
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(1, 49) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(1, 49) Warning: imported and not used: 'future' [UnusedImport]
/home/judge/data/code/Main.nim(1, 39) Warning: imported and not used: 'strscans' [UnusedImport]
/home/judge/data/code/Main.nim(1, 28) Warning: imported and not used: 'algorithm' [UnusedImport]

ソースコード

diff # 

import strutils, sequtils, algorithm, strscans, future

{.warning[SmallLshouldNotBeUsed]: off.}

const MOD = 1000000007

proc minus1(a: var string): string =
  for i in countDown(a.len-1, 0):
    if a[i] != '0':
      a[i] = ((a[i].int - '0'.int) - 1 + '0'.int).char
      break
    else:
      a[i] = '9'
  while a.len > 1 and a[0] == '0':
    a = a[1..a.len-1]
  result = a

proc solve(a: string, dp: var seq[seq[seq[seq[seq[int]]]]], pp: int): int =
  dp[0][0][0][0][0] = 1

  for i in 0..<a.len:
    for j in 0..1:
      for k in 0..1:
        for l in 0..2:
          var mlim = if a.len - i < 5: pp-1 else: 0
          for m in 0..mlim:
            for n in 0..9:
              if j == 0 and n > (a[i].int - '0'.int): break
              var
                p = if n < (a[i].int - '0'.int): 1 else: j
                q = if n == 3: 1 else: k
                r = if mlim == 0: 0 else: (n + m*10) mod pp
              dp[i+1][p][q][(n + l*10) mod 3][r] =
                  (dp[i+1][p][q][(n + l*10) mod 3][r] + dp[i][j][k][l][m]) mod MOD
  for j in 0..1:
    for k in 0..1:
      for l in 0..2:
        for m in 0..pp-1:
          if (k == 1 or l == 0) and m != 0:
            result = (result + dp[a.len][j][k][l][m]) mod MOD

when isMainModule:
  var
    input = stdin.readLine.split
    (a, b, p) = (input[0].minus1, input[1], input[2].parseint)
    dpA = newSeqWith(a.len+1, newSeqWith(2, newSeqWith(2, newSeqWith(3, newSeq[int](p)))))
    dpB = newSeqWith(b.len+1, newSeqWith(2, newSeqWith(2, newSeqWith(3, newSeq[int](p)))))

  var
    ansA = solve(a, dpA, p)
    ansB = solve(b, dpB, p)

  echo((ansB - ansA + MOD) mod MOD)
0