Submission #38235338

---

Source Code Expand

import Control.Monad
import Data.Array.IO
import Data.Array

main = do
  n <- readLn
  as <- map read . words <$> getLine
  ss <- replicateM n getLine
  arr <- abc286ep n as ss
  let aa = listArray (1,n) as
  q <- readLn
  replicateM_ q $ do
    [u,v] <- map read . words <$> getLine
    ans <- abc286em arr aa u v
    putStrLn ans

type DA = IOArray (Int,Int) (Int,Int) -- 距離の表

abc286ep :: Int       -- N
         -> [Int]     -- Ai
         -> [String]  -- Si
         -> IO DA        -- 距離の表
abc286ep n as ss = do
      arr <- newListArray ((1,1),(n,n)) $
        [ if sij == 'N' then (maxBound, 0) else (1, - ai)
        | (i,si,ai) <- zip3 [1..] ss as
        , (j,sij)   <- zip  [1..] si
        ]
      forM_ [1..n] (\k ->
        forM_ [1..n] (\i -> do
          dik <- readArray arr (i,k)
          when (dik < (maxBound, 0)) (
            forM_ [1..n] (\j -> do
              dkj <- readArray arr (k,j)
              when (dkj < (maxBound, 0)) $ do
                let dikj = add dik dkj
                dij <- readArray arr (i,j)
                when (dij > dikj) $ writeArray arr (i,j) dikj
              )
            )
          )
        )
      return arr

add (a,b) (c,d) = (a+c,b+d)

-- クエリ対応
abc286em :: DA            -- 距離の表
         -> Array Int Int -- Ai
         -> Int           -- Ui
         -> Int           -- Vi
         -> IO String        -- 答え
abc286em arr aa i j = do
  (k,a) <- readArray arr (i,j)
  return $ if k == maxBound then "Impossible" else unwords [show k, show $ aa ! j - a]

Submission Info

Compile Error

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

Judge Result