{-# LANGUAGE OverloadedStrings #-} {-# LANGUAGE BangPatterns #-} import qualified Control.Arrow as Arrow import qualified Data.ByteString.Char8 as BSC8 import qualified Data.Vector as V import qualified Data.Vector.Unboxed as VU import Data.Bits powMod :: Int -> Int -> Int -> Int powMod a 0 m = 1 powMod a n m | n `mod` 2 == 0 = ret * ret `mod` m where ret = powMod a (n `div` 2) m powMod a n m = powMod a (n-1) m * a `mod` m millerRabin :: Int -> Bool millerRabin n | n <= 1 = False | n == 2 || n == 3 || n == 5 || n == 7 = True | even n = False | otherwise = mrCheck n mrCheck :: Int -> Bool mrCheck n = loop [2,325,9375,28178,450775,9780504,1795265022] where !m = n - 1 !s = countTrailingZeros m !d = m `div` s loop [] = True loop (a:as) | powMod a d n /= 1 && powLoop 0 = False | otherwise = loop as where powLoop r | r < s = (powMod a (2 ^ r * d) n) /= (n - 1) && powLoop (r + 1) | otherwise = True type Parser a = BSC8.ByteString -> Maybe (a, BSC8.ByteString) parseInt :: Parser Int parseInt = fmap (Arrow.second BSC8.tail) . BSC8.readInt parseChar :: [Char] -> VU.Vector Char parseChar = VU.fromList parse1 :: IO Int parse1 = readLn parse2 :: IO (Int, Int) parse2 = (\vec -> (vec VU.! 0, vec VU.! 1)) . VU.unfoldrN 2 parseInt <$> BSC8.getLine parse3 :: IO (Int, Int, Int) parse3 = (\vec -> (vec VU.! 0, vec VU.! 1, vec VU.! 2)) . VU.unfoldrN 3 parseInt <$> BSC8.getLine parse4 :: IO (Int, Int, Int, Int) parse4 = (\vec -> (vec VU.! 0, vec VU.! 1, vec VU.! 2, vec VU.! 3)) . VU.unfoldrN 4 parseInt <$> BSC8.getLine parseM :: Int -> IO (VU.Vector Int) parseM m = VU.unfoldrN m parseInt <$> BSC8.getLine parseN :: Int -> IO (VU.Vector Int) parseN n = VU.replicateM n parse1 parseNM :: Int -> Int -> IO (V.Vector (VU.Vector Int)) parseNM n m = V.replicateM n $ VU.unfoldrN m parseInt <$> BSC8.getLine main :: IO () main = do n <- parse1 xs <- parseN n VU.mapM_ (BSC8.putStr . solve) xs solve :: Int -> BSC8.ByteString solve n | millerRabin n = tupToBS (n, 1) | otherwise = tupToBS (n, 0) where tupToBS (a, b) = BSC8.pack $ show a ++ " " ++ show b ++ "\n"