結果

問題 No.805 UMG
コンテスト
ユーザー kou_kkk
提出日時 2025-09-01 21:11:10
言語 Haskell
(9.10.1)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 693 bytes
コンパイル時間 11,894 ms
コンパイル使用メモリ 183,620 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2025-09-01 21:11:24
合計ジャッジ時間 8,710 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.10.1/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

module Main where

import Control.Monad (guard)
import Data.Array.Unboxed (listArray, (!), UArray)

main :: IO ()
main = print =<< solve <$> readLn <*> getLine

solve :: Int -> [Char] -> Integer
solve n s = let (us, ms) = foldr step ([], []) $ zip s [0 ..]
                charArr = listArray (0, n-1) s :: UArray Int Char
            in  sum $ umg charArr n us ms

step :: (Char, Int) -> ([Int], [Int]) -> ([Int], [Int])
step ('U', i) (u, m) = (i:u, m)
step ('M', i) (u, m) = (u, i:m)
step _ acc = acc

umg :: UArray Int Char -> Int -> [Int] -> [Int] -> [Integer]
umg a n us ms = do
  i <- us
  j <- ms
  guard (i < j)
  let k = i + (j - i) * 2
  guard (k < n)
  guard (a!k == 'G')
  return 1
0