Submission #19501713

---

Source Code Expand

{-# LANGUAGE DatatypeContexts #-}
import           Control.Exception           (assert)
import           Control.Monad
import           Control.Monad.Primitive
import qualified Control.Monad.ST            as ST
import qualified Data.Array.IO               as IO
import qualified Data.Array.ST               as ST
import qualified Data.Array.Unboxed          as A
import           Data.Bits
import qualified Data.ByteString.Char8       as BS
import           Data.Char
import           Data.Foldable
import           Data.List
import qualified Data.Map.Strict             as Map
import           Data.Maybe
import qualified Data.Sequence               as Seq
import qualified Data.Set                    as Set
import qualified Data.Vector                 as VB
import qualified Data.Vector.Mutable         as VBM
import qualified Data.Vector.Unboxed         as V
import           Data.Vector.Unboxed.Base
import qualified Data.Vector.Unboxed.Mutable as VM
import           Debug.Trace

readInt = fst . fromJust . BS.readInt
readIntList = map readInt . BS.words
getInt = readInt <$> BS.getLine
getIntList = readIntList <$> BS.getLine
getIntVec n = V.unfoldrN n (BS.readInt . BS.dropWhile isSpace) <$> BS.getLine

readInteger = fst . fromJust . BS.readInteger
readIntegerList = map readInteger . BS.words
getInteger = readInteger <$> BS.getLine
getIntegerList = readIntegerList <$> BS.getLine

inf :: Int
inf = 10^18

main = do
  [h, w, k] <- getIntList

  ss <- VB.replicateM h $ BS.getLine
  let getC y x | y >= 0 && x >= 0 = digitToInt $ (ss VB.! y) `BS.index` x
               | otherwise = 0

  let popc bm res | bm == 0 = res
                  | bm .&. 1 == 1 = popc (bm `shiftR` 1) (res+1)
                  | otherwise = popc (bm `shiftR` 1) res

      getRow i bm = getRow' i bm 0 0 []
      getRow' i bm j c res | j >= h = (c:res)
                           | otherwise = let bj = (bm `shiftR` j) .&. 1
                                             cji = getC j i
                                             c' = c + cji
                                         in if bj == 1
                                            then getRow' i bm (j+1) 0 (c':res)
                                            else getRow' i bm (j+1) c' res

      addRow []     []     res = Just (reverse res)
      addRow (x:xs) (y:ys) res | x + y <= k = addRow xs ys ((x+y):res)
                               | otherwise = Nothing

      solve' bm crow i res ubound
        | i >= w = res
        | otherwise = let nrow = getRow i bm
                          over = any (> k) nrow
                          crow' = addRow crow nrow []
                      in case () of
                           () | over -> inf
                              | isJust crow' -> solve' bm (fromJust crow') (i+1) res ubound
                              | res <= ubound -> solve' bm nrow (i+1) (res+1) ubound
                              | otherwise -> inf

      solve :: Int -> Int -> Int
      solve bm res
        | bm < 0 = res
        | otherwise = let r = solve' bm (getRow (-1) bm) 0 (popc bm 0) res
                          res' = min r res
                      in solve (bm-1) res'

  print $ solve (1 `shiftL` (h-1)) inf

Submission Info

Compile Error

Loaded package environment from /home/contestant/.ghc/x86_64-linux-8.8.3/environments/default

Main.hs:1:14: warning:
    -XDatatypeContexts is deprecated: It was widely considered a misfeature, and has been removed from the Haskell language.
  |
1 | {-# LANGUAGE DatatypeContexts #-}
  |              ^^^^^^^^^^^^^^^^

Judge Result