{-# 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 import GHC.Integer.GMP.Internals fi :: Int -> Integer fi = fromIntegral fI :: Integer -> Int fI = fromInteger powModInt :: Int -> Int -> Int -> Int powModInt !a !n !mo = fI $ powModInteger (fi a) (fi n) (fi mo) millerRabin :: Int -> Bool millerRabin n | n <= 1 = False | n == 2 || n == 3 || n == 5 || n == 7 = True | n `mod` 2 == 0 = False | otherwise = mrCheck n mrCheck :: Int -> Bool mrCheck n = loop [2,325,9375,28178,450775,9780504,1795265022] where !s = countTrailingZeros (n - 1) !d = (n - 1) `div` s loop [] = True loop (a:as) | (powModInt a d n) /= 1 && powLoop 0 = False | otherwise = loop as where powLoop r | r < s = (powModInt 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"