結果

問題 No.286 Modulo Discount Store
ユーザー aimyaimy
提出日時 2017-05-13 10:16:20
言語 Haskell
(9.8.2)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 722 bytes
コンパイル時間 5,058 ms
コンパイル使用メモリ 148,832 KB
最終ジャッジ日時 2024-09-15 12:14:19
合計ジャッジ時間 5,564 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge6
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
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:1:1: error: [GHC-87110]
    Could not load module ‘Data.IntMap’.
    It is a member of the hidden package ‘containers-0.6.8’.
    Use -v to see a list of the files searched for.
  |
1 | import Data.IntMap ((!))
  | ^^^^^^^^^^^^^^^^^^^^^^^^

Main.hs:2:1: error: [GHC-87110]
    Could not load module ‘Data.IntMap’.
    It is a member of the hidden package ‘containers-0.6.8’.
    Use -v to see a list of the files searched for.
  |
2 | import qualified Data.IntMap as M
  | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

ソースコード

diff # 

import Data.IntMap ((!))
import qualified Data.IntMap as M
import Control.Monad
import Data.Bits

main = do
 n <- readLn
 ms <- map read . words <$> getContents
 print (mds n ms)

mds :: Int -> [Int] -> Int
mds n ms = snd $ M.findMax $ foldr (dp n ms t) (M.singleton 0 0) (replicate n [0..n-1])
 where 
  t = M.fromDistinctAscList $ do
   let es = map bit [0..n]
   bs <- replicateM n [0,1]
   let k = sum (zipWith (*) (reverse bs) es)
   let v = sum (zipWith (*) (reverse bs) ms) `mod` 1000
   return (k,v)

dp n ms t ds im = M.unionsWith min $ map buy ds
 where
  buy d = M.mapKeysMonotonic (flip setBit d) $ M.mapWithKey (\k v -> v + (max 0 ((ms!!d) - (t!k)))) $
          M.filterWithKey (\k _ -> not (testBit k d)) im
0