結果

問題 No.4 おもりと天秤
ユーザー myuon
提出日時 2015-03-09 09:32:00
言語 Haskell
(9.10.1)
結果
WA  
実行時間 -
コード長 618 bytes
コンパイル時間 6,335 ms
コンパイル使用メモリ 225,536 KB
実行使用メモリ 17,152 KB
最終ジャッジ日時 2024-06-24 16:48:48
合計ジャッジ時間 5,509 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other WA * 23
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 Control.Applicative
import Data.List
import Data.Array

solve ws = go ws arr where
  arr = listArray ((0,0),(length ws, sum ws `div` 2 + 1)) $ True : repeat False
  check m = (/= []) $ filter (\((_,w),v) -> 2 * w == sum ws && v == True) $ assocs m

  go [] m = check m
  go (x:xs) m
    | check m = True
    | otherwise = let m' = accum (||) m $ filter (\((k,w),_) -> 2 * w <= sum ws) $ fmap (\((k,w),v) -> ((k+1,w+x),v)) $ filter (\(_,v) -> v == True) $ assocs m in go xs m'

main = do
  _ <- getLine
  ws <- fmap (read :: String -> Int) . words <$> getLine
  print $ solve $ sortBy (\x y -> compare y x) $ ws
0