結果

問題 No.78 クジ付きアイスバー
コンテスト
ユーザー ducktail
提出日時 2018-09-03 21:01:47
言語 Haskell
(9.14.1)
コンパイル:
ghc -rtsopts -with-rtsopts=-K1G -o a.out -O2 _filename_
実行:
./a.out
結果
AC  
実行時間 3 ms / 5,000 ms
コード長 708 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 9,624 ms
コンパイル使用メモリ 193,536 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2026-04-25 12:08:44
合計ジャッジ時間 7,437 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.14.1/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.Internal.List):
    "This is a partial function, it throws an error on empty lists. Use pattern matching, 'Data.List.uncons' or 'Data.Maybe.listToMaybe' 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.Internal.List):
    "This is a partial function, it throws an error on empty lists. Replace it with 'drop' 1, or use pattern matching or 'GHC.Internal.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 #
raw source code 

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