結果

問題 No.197 手品
ユーザー くれちー
提出日時 2017-02-22 16:36:01
言語 Haskell
(9.10.1)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 852 bytes
コンパイル時間 2,041 ms
コンパイル使用メモリ 174,080 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 03:51:15
合計ジャッジ時間 3,443 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 43
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff #

import Data.List (elemIndex)

data Result = SUCCESS | FAILURE
    deriving (Show)

count :: Eq a => a -> [a] -> Int
count n xs = length $ filter (\x -> x == n) xs

solve :: Integral a => String -> a -> String -> Result
solve sb n sa
    | ob /= oa      = SUCCESS
    | bd == Nothing = FAILURE
    | bd == Just 1 && ad == Just 1 = if n == 1
                 then SUCCESS
                 else FAILURE
    | d > n         = SUCCESS
    | otherwise     = FAILURE
    where
        ob     = count 'o' sb
        oa     = count 'o' sa
        xb     = count 'x' sb
        xa     = count 'x' sa
        c      = if ob < xb then 'o' else 'x'
        bd     = elemIndex c sb
        ad     = elemIndex c sa
        Just d = fmap (fromIntegral . abs) $ (-) <$> bd <*> ad

main = do
    sb <- getLine
    n  <- readLn
    sa <- getLine
    print $ solve sb n sa
0