結果

問題 No.59 鉄道の旅
ユーザー aimyaimy
提出日時 2017-06-18 11:34:32
言語 Haskell
(9.6.2)
結果
AC  
実行時間 1,103 ms / 5,000 ms
コード長 1,441 bytes
コンパイル時間 3,067 ms
コンパイル使用メモリ 179,588 KB
実行使用メモリ 150,088 KB
最終ジャッジ日時 2023-07-25 00:25:54
合計ジャッジ時間 18,156 ms
ジャッジサーバーID
(参考情報)
judge12 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 792 ms
142,000 KB
testcase_01 AC 783 ms
141,868 KB
testcase_02 AC 793 ms
141,976 KB
testcase_03 AC 792 ms
142,028 KB
testcase_04 AC 1,103 ms
150,088 KB
testcase_05 AC 782 ms
141,912 KB
testcase_06 AC 772 ms
141,888 KB
testcase_07 AC 784 ms
134,708 KB
testcase_08 AC 812 ms
141,912 KB
testcase_09 AC 813 ms
138,936 KB
testcase_10 AC 803 ms
130,684 KB
testcase_11 AC 774 ms
130,680 KB
testcase_12 AC 782 ms
130,656 KB
testcase_13 AC 814 ms
105,604 KB
testcase_14 AC 825 ms
105,116 KB
testcase_15 AC 714 ms
97,848 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff #

import Data.List
import qualified Data.ByteString.Char8 as B
import Data.Maybe

data SegTree a = Leaf a | Node a (SegTree a) (SegTree a) deriving (Eq, Show)

val :: SegTree a -> a
val (Leaf v) = v
val (Node v _ _) = v

fromList :: (Num a) => Int -> [a] -> SegTree a
fromList 1 [x] = Leaf x
fromList n xs  = Node (val left + val right) left right
 where
  n' = div n 2
  (xs1,xs2) = splitAt n' xs
  left = fromList n' xs1
  right = fromList (n-n') xs2

update :: (Num a) => Int -> SegTree a -> Int -> a -> SegTree a
update 1 (Leaf v) 1 x = Leaf (v+x)
update n (Node _ l r) i x
 | i <= n' = Node (val left + val r) left r
 | otherwise = Node (val l + val right) l right
 where
  n' = div n 2
  left = update n' l i x
  right = update (n-n') r (i-n') x

query :: (Num a) => Int -> SegTree a -> Int -> Int -> a
query 1 (Leaf v) 1 1 = v
query n (Node v l r) i j
 | (i,j) == (1,n) = v
 | j <= n' = query n' l i j
 | i > n' = query (n-n') r (i-n') (j-n')
 | otherwise = query n' l i n' + query (n-n') r 1 (j-n')
 where n' = div n 2

maxv = 1000000

main = do
 [_,k] <- map read . words <$> getLine
 ws <- map (fst . fromJust . B.readInt) . B.words <$> B.getContents
 let seg = fromList maxv (replicate maxv 0)
 print (freight k seg ws)

freight k seg = val . foldl' (dp k) seg

dp k seg w
 | w < 0 && query maxv seg (abs w) (abs w) /= 0 = update maxv seg (abs w) (-1)
 | w > 0 && query maxv seg w maxv < k = update maxv seg w 1
 | otherwise = seg
0