提出 #17917533

---

ソースコード 拡げる

{-# LANGUAGE FlexibleContexts #-}

import qualified Data.ByteString.Char8 as C
import Data.List
import qualified Data.IntMap.Strict as M
import qualified Data.IntSet as S
import qualified Data.Vector as V
import qualified Data.Vector.Unboxed as U

main = sub =<< get

sub [n,c] = sol n c <$> get3N n >>= print

get = unfoldr (C.readInt . C.dropWhile (<'+')) <$> C.getLine

get3N n = V.replicateM n $ U.unfoldrN 3 (C.readInt . C.dropWhile (<'+')) <$> C.getLine

sol n c = maximum . scanl1 (+)
  . fmap snd . M.toAscList
  . M.foldrWithKey' h M.empty
  . M.filter (not . S.null) . M.map g
  . V.foldr f M.empty

f stc = M.insertWith S.union t (S.singleton (-c)) . M.insertWith S.union s (S.singleton c)
  where
  s = stc U.! 0
  t = stc U.! 1
  c = stc U.! 2

g cs = S.foldr' g' cs cs
  where
  g' c s = if c>0 && S.member (-c) s then S.delete c $ S.delete (-c) s else s

h k cs m = S.foldr' h' m cs :: M.IntMap Int
  where
  h' c = if c>0 then M.insertWith (+) (k-1) 1 else M.insertWith (+) k (-1)

提出情報

コンパイルエラー

Loaded package environment from /home/contestant/.ghc/x86_64-linux-8.8.3/environments/default

ジャッジ結果