Submission #10728035

---

Source Code Expand

import           Control.Monad
import qualified Data.ByteString.Char8       as BS
import           Data.Char
import           Data.Maybe
import qualified Data.Vector.Unboxed.Mutable as VM

readInt = fst . fromJust . BS.readInt
readIntList = map readInt . BS.words
getInt = readInt <$> BS.getLine
getIntList = readIntList <$> BS.getLine

main = do
  [n, p] <- getIntList
  s <- BS.getLine

  v <- VM.new p
  VM.set v (0::Int)
  VM.write v 0 1 -- U0 `mod` p == 0                                                                                       
  let solve25 i a | i > n = print a
                  | otherwise = do
                      let c = BS.index s (n-i)
                          di = ord c - ord '0'
                          x = if di `mod` p == 0 then n - i + 1 else 0
                      solve25 (i+1) (a+x)

      solve i a md m10 | i > n = print a
                       | otherwise = do
                           let c = BS.index s (n-i)
                               di = ord c - ord '0'
                               md' = (md + di * m10) `mod` p
                               m10' = (m10 * 10) `mod` p
                           y <- VM.read v md'
                           VM.write v md' (y+1)
                           solve (i+1) (a+y) md' m10'

  if 10 `mod` p == 0
    then solve25 1 0
    else solve 1 0 0 1

Submission Info

Judge Result