結果

問題 No.488 四角関係
ユーザー ducktailducktail
提出日時 2018-08-24 14:04:38
言語 Haskell
(9.10.1)
結果
AC  
実行時間 108 ms / 5,000 ms
コード長 940 bytes
コンパイル時間 8,853 ms
コンパイル使用メモリ 226,176 KB
実行使用メモリ 8,320 KB
最終ジャッジ日時 2024-06-12 04:15:02
合計ジャッジ時間 10,302 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default
[1 of 2] Compiling Main             ( Main.hs, Main.o )
[2 of 2] Linking a.out

ソースコード

diff # 

import Control.Applicative
import Control.Monad
import Data.Vector (Vector, (!), (//))
import qualified Data.Vector as V
import Data.Vector.Mutable (IOVector, STVector)
import qualified Data.Vector.Mutable as VM

main :: IO ()
main = do
  [n, m] <- f
  ev <- do
    v <- VM.replicate n [] :: IO (IOVector [Int])
    replicateM_ m $ do
      [i, j] <- f
      xs <- VM.read v i
      VM.write v i (j:xs)
      ys <- VM.read v j
      VM.write v j (i:ys)
    V.freeze v
  print $ solve n ev
  where
    f = map read <$> words <$> getLine

solve :: Int -> Vector [Int] -> Int
solve n ev = length $ do
  i <- [0 .. n-4]
  j <- [i+1 .. n-3]
  k <- [j+1 .. n-2]
  l <- [k+1 .. n-1]
  guard $ (sum . map (egn i)) [j, k, l] == 2
  guard $ (sum . map (egn j)) [i, k, l] == 2
  guard $ (sum . map (egn k)) [i, j, l] == 2
  guard $ (sum . map (egn l)) [i, j, k] == 2
  return ()
  where
    egn x y
      | x `elem` (ev ! y) = 1
      | otherwise = 0
0