結果

問題 No.216 FAC
コンテスト
ユーザー poapoa
提出日時 2020-08-10 15:55:49
言語 Haskell
(9.14.1)
コンパイル:
ghc -rtsopts -with-rtsopts=-K1G -o a.out -O2 _filename_
実行:
./a.out
結果
AC  
実行時間 3 ms / 1,000 ms
コード長 1,175 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,183 ms
コンパイル使用メモリ 208,008 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2026-04-24 07:34:17
合計ジャッジ時間 4,527 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
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:22:13: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Prelude, 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"."
   |
22 |     then if head ys == 0 then putStrLn "YES" else putStrLn "NO"
   |             ^^^^

Main.hs:26:41: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Prelude, 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"."
   |
26 |                   isK     = 0 == (snd . head $ xs)
   |                                         ^^^^

Main.hs:27:35: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘head’
    (imported from Prelude, 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"."
   |
27 |                   kScore  = fst . head $ xs
   |                                   ^^^^

Main.hs:28:71: warning: [GHC-63394] [-Wx-partial]
    In the use of ‘tail’
    (imported from Prelude, 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"."
   |
28 |                   others  = fst $ List.maximumBy (Ord.comparing fst) (tail xs)
   |                                                                       ^^^^

[2 o

ソースコード

diff #
raw source code 

import           Control.Arrow         ((&&&))
import qualified Data.Char             as Char
import qualified Data.List             as List
import qualified Data.ByteString.Char8 as BSC8
import qualified Data.Bool             as Bool
import qualified Data.Ord              as Ord

getInt :: IO [Int]
getInt = List.unfoldr f <$> BSC8.getLine
  where
    -- f :: BSC8.ByteString -> Maybe (Int, BSC8.ByteString)
    f s = do
      (n, s') <- BSC8.readInt s
      return (n, BSC8.dropWhile Char.isSpace s')

main :: IO ()
main = do
  n <- readLn :: IO Int
  xs <- getInt
  ys <- getInt
  if n == 1
    then if head ys == 0 then putStrLn "YES" else putStrLn "NO"
    else putStrLn $ Bool.bool "NO" "YES" $ solver n (zip xs ys)
solver n zs = let
                  xs      = map (f &&& g) $ List.groupBy (\a b -> snd a == snd b) $ List.sortBy (Ord.comparing snd) zs
                  isK     = 0 == (snd . head $ xs)
                  kScore  = fst . head $ xs
                  others  = fst $ List.maximumBy (Ord.comparing fst) (tail xs)
              in  isK && (kScore >= others)
f []          = 0
f ((a, b):[]) = a
f ((a, b):xs) = a + f xs

g []          = 0
g ((a, b):xs) = b
0