結果

問題 No.914 Omiyage
ユーザー torus711
提出日時 2019-10-26 15:52:18
言語 Haskell
(9.10.1)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,357 bytes
コンパイル時間 5,556 ms
コンパイル使用メモリ 182,528 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-14 21:48:11
合計ジャッジ時間 6,273 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
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:36:1: warning: [GHC-94817] [-Wtabs]
    Tab character found here, and in 34 further locations.
    Suggested fix: Please use spaces instead.
   |
36 |         [ n, m, k ] <- readInts
   | ^^^^^^^^
[2 of 2] Linking a.out

ソースコード

diff # 

{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE BinaryLiterals #-}
{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE NumDecimals #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TupleSections #-}

import Control.Applicative
import Control.Arrow
import Control.Monad
import Control.Monad.ST

import Data.Char
import Data.List
import Data.Maybe

import qualified Data.ByteString.Char8 as B

import Data.Array.ST.Safe
import Data.STRef

import Debug.Trace
import Text.Printf

readInt = readLn :: IO Int
readInteger = readLn :: IO Integer
readInts = map ( fst . fromJust . B.readInt ) . B.words <$> B.getLine
readIntegers = map ( fst . fromJust . B.readInteger ) . B.words <$> B.getLine

which a b f = if f then a else b
mp [ a, b ] = ( a, b )

modifyArray a i f = writeArray a i =<< f <$> readArray a i

main = do
	[ n, m, k ] <- readInts
	as <- replicateM n readInts
	print $ solve n m k as

solve n m k as = runST $ do
	dp <- newArray_ ( ( 0, 0 ), ( n, k ) ) :: ST s ( STUArray s ( Int, Int ) Bool )
	writeArray dp ( 0, k ) True
	forM_ [ 0 .. n - 1 ] $ \i -> do
		forM_ [ 0 .. k ] $ \j ->do
			forM_ ( as !! i ) $ \a -> do
				when ( 0 <= j - a ) $ do
					b <- readArray dp ( i, j )
					modifyArray dp ( i + 1, j - a ) ( || b )
	res <- newSTRef ( -1 )
	forM_ [ k, k - 1 .. 0 ] $ \j -> do
		b <- readArray dp ( n, j )
		when b $ do
			writeSTRef res j
	readSTRef res
0