
はじめに
コード例が Haskell になっている記事がちょこちょこあるのにも関わらず Haskell 用のテンプレをブログ内に用意していなかったので,そろそろ書いておくか,という感じのアレです.
一応セクションごとにコメントを付しますが,よく分からずに Haskell を書いているのでツッコミどころがあるかもしれません.マサカリを投げる際はゆっくり投擲してくださると助かります.
テンプレ全容
とりあえず全容を載せておきます.
{-# LANGUAGE BinaryLiterals #-}
{-# LANGUAGE NegativeLiterals #-}
{-# LANGUAGE HexFloatLiterals #-}
{-# LANGUAGE NumericUnderscores #-}
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE TupleSections #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE BlockArguments #-}
{-# LANGUAGE NumDecimals #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE TypeApplications #-}
{-# OPTIONS_GHC -O2 #-}
{-# OPTIONS_GHC -Wno-tabs #-}
{-# LANGUAGE CPP #-}
#define ATCODER
import Control.Applicative
import Control.Arrow
import Control.Monad
import Control.Monad.ST
import Data.Bool
import Data.Bits
import Data.Char
import Data.List hiding ( (!?) )
import Data.Maybe
import Data.Either
import Data.Function
import qualified Data.ByteString.Char8 as B
import Data.Array (Array)
import Data.Array.IArray
import Data.Array.Unboxed (UArray)
import Data.Array.ST.Safe
import Data.Array.MArray
import Data.STRef
import Text.Printf
import Debug.Trace
#ifdef ATCODER
import Control.Monad.Extra
import Data.Tuple.Extra hiding ( (&&&), (***) )
import Data.List.Extra hiding ( (!?) )
import qualified Data.Set as Set
import qualified Data.IntSet as ISet
import qualified Data.Map as Map
import qualified Data.IntMap as IMap
import qualified Data.Sequence as Seq
import Data.Sequence ( (<|), (|>), (><), ViewL(..), ViewR(..) )
#endif
readInt = fst . fromJust . B.readInt <$> B.getLine
readInteger = fst . fromJust . B.readInteger <$> B.getLine
readInts = map ( fst . fromJust . B.readInt ) . B.words <$> B.getLine
readIntegers = map ( fst . fromJust . B.readInteger ) . B.words <$> B.getLine
readStrs = map B.unpack . B.words <$> B.getLine
yesno = bool "No" "Yes"
mp [ a, b ] = ( a, b )
count a as = length $ filter ( == a ) as
countIf f as = length $ filter f as
swapArrayElem a i j = do
x <- readArray a i
y <- readArray a j
writeArray a i y
writeArray a j x
a !? i
| inRange ( bounds a ) i = Just $ a ! i
| otherwise = Nothing
printList :: Show a => [a] -> IO ()
printList = putStrLn . intercalate " " . map show
#ifdef ATCODER
readStr = trim . B.unpack <$> B.getLine
printList2d :: Show a => [[a]] -> IO ()
printList2d = mapM_ printList
printArray2d ass = printList2d $ chunksOf ( x2 - x1 + 1 ) $ elems ass
where
( ( _, x1 ), ( _, x2 ) ) = bounds ass
#endif
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE BinaryLiterals #-}
{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE NumDecimals #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TupleSections #-}
{-# OPTIONS_GHC -O2 #-}
{-# OPTIONS_GHC -Wno-tabs #-}
便利そうだと思った GHC 拡張をつらつら書いています.あんまり使っていないという説もありますが…….
あと,提出コードよりコンパイルコマンドの方を簡潔にしたいのでコンパイルオプションもコード中に書いています.
import たち
import Control.Applicative
import Control.Arrow
import Control.Monad
import Control.Monad.ST
import Data.Char
import Data.List
import Data.Maybe
import qualified Data.ByteString.Char8 as B
import Data.Array
import Data.Array.ST.Safe
import Data.STRef
import Debug.Trace
import Text.Printf
見たまんまで,よく使いそうなモジュールを import しまくっています.最近は Data.Set, Data.Map, Data.Sequence あたりを加えるかどうかで迷っています.
入力関係
readInt = readLn :: IO Int
readInteger = readLn :: IO Integer
readInts = map ( fst . fromJust . B.readInt ) . B.words <$> B.getLine
readIntegers = map ( fst . fromJust . B.readInteger ) . B.words <$> B.getLine
入力で readLn をよく使いますが,Int 用と Integer 用で別名を付けて使い分けています.
また,よく知られているように長い入力を String で処理すると大変なことになるので,Data.ByteString を介したリストの入力を,これまた Int 用と Integer 用に分けて用意しています.
ユーティリティ的な
which a b f = if f then a else b
mp [ a, b ] = ( a, b )
if が式なのがちょっと使いづらくて,関数っぽく使いたいので if をくるむ関数を用意しています.ただ,Data.Bool に同等の関数があることに気付いてしまいました.引数の順序が逆なのがちょっとイヤなのでそのままにしています.
mp は,リストで受け取った $2$ つの値をタプルに変換できると,uncurry した関数に渡せるようになってたまに便利なので用意しています.
modifyArray
modifyArray a i f = writeArray a i =<< f <$> readArray a i
STRef に対する modifySTRef の Array 版です.Hoogle を検索すると出てきますが,「無い」としている記事も見つかるので,よく分かりませんがこのままにしておきます.
printList
printList [] = putChar '\n'
printList [a] = print a
printList (a:as) = putStr ( show a ) >> putChar ' ' >> printList as
たまに要求される,リストの空白区切り出力です.空リストは一旦放置で(出題頻度が低くて,書くのもすぐなので).