結果

問題 No.78 クジ付きアイスバー
コンテスト
ユーザー ducktail
提出日時 2018-09-03 21:01:47
言語 Haskell
(9.10.1)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 708 bytes
コンパイル時間 10,426 ms
コンパイル使用メモリ 171,648 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-09 09:28:19
合計ジャッジ時間 4,117 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )

Main.hs:25:33: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Data.List, but defined in GHC.List):
    "This is a partial function, it throws an error on empty lists. Use pattern matching or Data.List.uncons instead. Consider refactoring to use Data.List.NonEmpty."
   |
25 |       | cat == 0 = f (cib + 1) (head xs) (tail xs)
   |                                 ^^^^

Main.hs:25:43: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘tail’
    (imported from Data.List, but defined in GHC.List):
    "This is a partial function, it throws an error on empty lists. Replace it with drop 1, or use pattern matching or Data.List.uncons instead. Consider refactoring to use Data.List.NonEmpty."
   |
25 |       | cat == 0 = f (cib + 1) (head xs) (tail xs)
   |                                           ^^^^
[2 of 2] Linking a.out

ソースコード

diff # 

import Control.Applicative
import Data.List
import Data.Char (digitToInt)

main :: IO ()
main = solve <$> f <*> getLine >>= print
  where
    f = map read <$> words <$> getLine

solve :: [Int] -> String -> Int
solve [n, k] s
  | q == 0 = fst . ice 0 $ take r is
  | cb <= cat = cb
  | otherwise = cb + (q-1) * (cb - cat) + (fst . ice cat $ take r is)
  where
    is = map digitToInt s
    (q, r) = divMod k n
    (cb, cat) = ice 0 is

ice :: Int -> [Int] -> (Int, Int)
ice at ibs = f 0 at ibs
  where
    f cib cat [] = (cib, cat)
    f cib cat xs
      | cat == 0 = f (cib + 1) (head xs) (tail xs)
      | otherwise = let (as, bs) = splitAt cat xs
                    in f cib (cat - length as + sum as) bs
0