結果

問題 No.1417 100の倍数かつ正整数(2)
コンテスト
ユーザー dot_haraai
提出日時 2021-05-31 23:15:16
言語 Nim
(2.2.0)
結果
WA  
実行時間 -
コード長 3,040 bytes
コンパイル時間 5,363 ms
コンパイル使用メモリ 88,320 KB
実行使用メモリ 7,040 KB
最終ジャッジ日時 2024-11-08 23:59:43
合計ジャッジ時間 5,353 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24 WA * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
/home/judge/data/code/Main.nim(2, 18) Warning: Use the new 'sugar' module instead; future is deprecated [Deprecated]
/home/judge/data/code/Main.nim(1, 8) Warning: imported and not used: 'times' [UnusedImport]
/home/judge/data/code/Main.nim(2, 26) Warning: imported and not used: 'strformat' [UnusedImport]
/home/judge/data/code/Main.nim(2, 18) Warning: imported and not used: 'future' [UnusedImport]
/home/judge/data/code/Main.nim(2, 37) Warning: imported and not used: 'deques' [UnusedImport]
/home/judge/data/code/Main.nim(1, 41) Warning: imported and not used: 'algorithm' [UnusedImport]
/home/judge/data/code/Main.nim(1, 52) Warning: imported and not used: 'tables' [UnusedImport]
/home/judge/data/code/Main.nim(1, 60) Warning: imported and not used: 'sets' [UnusedImport]
/home/judge/data/code/Main.nim(1, 66) Warning: imported and not used: 'lists' [UnusedImport]
/home/judge/data/code/Main.nim(1, 73) Warning: imported and not used: 'intsets' [UnusedImport]
/home/judge/data/code/Main.nim(2, 8) Warning: imported and not used: 'critbits' [UnusedImport]

ソースコード

diff # 

import times, strutils, sequtils, math, algorithm, tables, sets, lists, intsets
import critbits, future, strformat, deques
template `max=`(x,y) = x = max(x,y)
template `min=`(x,y) = x = min(x,y)
template `mod=`(x,y) = x = x mod y
template scan2 = (scan(), scan())
template scan3 = (scan(), scan())
let read* = iterator: string {.closure.} =
    while true: (for s in stdin.readLine.split: yield s)
proc scan(): int = read().parseInt
proc scanf(): float = read().parseFloat
proc toInt(c:char): int =
    return int(c) - int('0')



proc solve():int=
  var
    s = read()
    n = s.len()
    md = 10^9+7
    dp = newseqwith(n,newseqwith(2,newseqwith(3,newseqwith(3,0))))
  # dp[i][1][j][k]:
  # i桁目まで見てs未満が確定していて、2がj個、5がk個存在する
  # ある桁から発生する場合：
  # すでにs未満は確定している

  for i in 1..<s[0].toInt:
    case i
      of 2,6:
        dp[0][1][1][0]=1
      of 4,8:
        dp[0][1][2][0]+=1
      of 5:
        dp[0][1][0][1]+=1
      else:
        dp[0][1][0][0] += 1
  var i = s[0].toInt
  case i
      of 2,6:
        dp[0][0][1][0]=1
      of 4,8:
        dp[0][0][2][0]+=1
      of 5:
        dp[0][0][0][1]+=1
      else:
        dp[0][0][0][0] += 1

  
  for digit in 1..<n:
    
    for i in 1..9:
      case i
        of 2,6:
          dp[digit][1][1][0]+=1
          for j in 0..2:
            for k in 0..2:
              dp[digit][1][min(2,j+1)][k]+=dp[digit-1][1][j][k]
              if i<s[digit].toInt:
                dp[digit][1][min(2,j+1)][k]+=dp[digit-1][0][j][k]
              elif i==s[digit].toInt:
                dp[digit][0][min(2,j+1)][k]+=dp[digit-1][0][j][k]
        of 4,8:
          dp[digit][1][2][0]+=1
          for j in 0..2:
            for k in 0..2:
              dp[digit][1][min(2,j+2)][k]+=dp[digit-1][1][j][k]
              if i<s[digit].toInt:
                dp[digit][1][min(2,j+2)][k]+=dp[digit-1][0][j][k]
              elif i==s[digit].toInt:
                dp[digit][0][min(2,j+2)][k]+=dp[digit-1][0][j][k]
        of 5:
          dp[digit][1][0][1]+=1
          for j in 0..2:
            for k in 0..2:
              dp[digit][1][j][min(2,k+1)]+=dp[digit-1][1][j][k]
              if i<s[digit].toInt:
                dp[digit][1][j][min(2,k+1)]+=dp[digit-1][0][j][k]
              elif i==s[digit].toInt:
                dp[digit][0][j][min(2,k+1)]+=dp[digit-1][0][j][k]
        else:
          # 2とも5とも関係ない
          # この桁から発生する場合
          dp[digit][1][0][0] += 1
          # 前の桁がある場合
          for j in 0..2:
            for k in 0..2:
              dp[digit][1][j][k]+=dp[digit-1][1][j][k]
              if i<s[digit].toInt:
                dp[digit][1][j][k]+=dp[digit-1][0][j][k]
              elif i==s[digit].toInt:
                dp[digit][0][j][k]+=dp[digit-1][0][j][k]
    for i in 0..1:
      for k in 0..2:
        for l in 0..2:
          dp[digit][i][k][l].mod=md
  return (dp[n-1][0][2][2]+dp[n-1][1][2][2]).mod(md)
  
echo solve()
0