結果
問題 | No.1232 2^x = x |
ユーザー | かりあげクン |
提出日時 | 2020-09-25 12:46:00 |
言語 | Haskell (9.8.2) |
結果 |
CE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 8,146 bytes |
コンパイル時間 | 257 ms |
コンパイル使用メモリ | 169,856 KB |
最終ジャッジ日時 | 2024-11-14 23:50:28 |
合計ジャッジ時間 | 731 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。(リジャッジ後のコンパイルエラーは公開されます)
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
ただし、clay言語の場合は開発者のデバッグのため、公開されます。
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.8.2/environments/default [1 of 2] Compiling Main ( Main.hs, Main.o ) Main.hs:52:1: error: [GHC-87110] Could not load module ‘GHC.Integer.GMP.Internals’. It is a member of the hidden package ‘integer-gmp-1.1’. Use -v to see a list of the files searched for. | 52 | import qualified GHC.Integer.GMP.Internals as GMP | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ Main.hs:53:1: error: [GHC-87110] Could not find module ‘GHC.Integer.Logarithms.Internals’. Use -v to see a list of the files searched for. | 53 | import qualified GHC.Integer.Logarithms.Internals as Log | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ソースコード
{-# LANGUAGE BangPatterns #-} {-# LANGUAGE CPP #-} {-# LANGUAGE DerivingStrategies #-} {-# LANGUAGE FlexibleContexts #-} {-# LANGUAGE GeneralizedNewtypeDeriving #-} {-# LANGUAGE MagicHash #-} {-# LANGUAGE MultiParamTypeClasses #-} {-# LANGUAGE RankNTypes #-} {-# LANGUAGE RecordWildCards #-} {-# LANGUAGE ScopedTypeVariables #-} {-# LANGUAGE TypeApplications #-} {-# LANGUAGE TypeFamilies #-} {-# LANGUAGE UnboxedTuples #-} {- base -} import Control.Arrow import Control.Monad import Control.Monad.ST import Data.Bits import Data.Bool import Data.Char import Data.Coerce import Data.Function import Data.Functor.Identity import Data.IORef import Data.Ix import Data.Monoid import Data.Ord import Data.Ratio import Data.Semigroup import Data.STRef import Data.Void import Data.Word import GHC.Exts import Unsafe.Coerce {- array -} import qualified Data.Array as Arr import qualified Data.Array.IO as ArrIO import qualified Data.Array.MArray as ArrMA import qualified Data.Array.ST as ArrST import qualified Data.Array.Unboxed as ArrU {- bytestring -} import qualified Data.ByteString as BS import qualified Data.ByteString.Builder as BSB import qualified Data.ByteString.Builder.Extra as BSBE import qualified Data.ByteString.Builder.Prim as BSBP import qualified Data.ByteString.Char8 as BSC8 import qualified Data.ByteString.Lazy as BSL import qualified Data.ByteString.Lazy.Char8 as BSLC8 import qualified Data.ByteString.Short as BSS import qualified Data.ByteString.Unsafe as BSU {- integer-gmp -} import qualified GHC.Integer.GMP.Internals as GMP import qualified GHC.Integer.Logarithms.Internals as Log {- vector -} import qualified Data.Vector as V import qualified Data.Vector.Fusion.Stream.Monadic as VFSM import qualified Data.Vector.Generic as VG import qualified Data.Vector.Generic.Mutable as VGM import qualified Data.Vector.Mutable as VM import qualified Data.Vector.Primitive as VP import qualified Data.Vector.Primitive.Mutable as VPM import qualified Data.Vector.Storable as VS import qualified Data.Vector.Storable.Mutable as VSM import qualified Data.Vector.Unboxed as VU import qualified Data.Vector.Unboxed.Mutable as VUM ------------------------------------------------------------------------------- #define MOD 1000000007 #define Fact_CACHE_SIZE 100100 #define LOG_FACT_CACHE_SIZE 1024 ------------------------------------------------------------------------------- fi :: Int -> Integer fi = fromIntegral {-# INLINE fi #-} fI :: Integer -> Int fI = fromInteger {-# INLINE fI #-} floorSqrt :: Int -> Int floorSqrt = floor . sqrt . fromIntegral {-# INLINE floorSqrt #-} floorLog2 :: Int -> Int floorLog2 x = fromIntegral $ unsafeShiftR y 52 - 1023 where y :: Word64 y = unsafeCoerce (fromIntegral x :: Double) {-# INLINE floorLog2 #-} powModInt :: Int -> Int -> Int -> Int powModInt a b c = fI $ GMP.powModInteger (fi a) (fi b) (fi c) {-# INLINE powModInt #-} recipModInt :: Int -> Int -> Int recipModInt a m = fI $ GMP.recipModInteger (fi a) (fi m) {-# INLINE recipModInt #-} gcdext :: Int -> Int -> (Int, Int, Int) -- a * x + b * y = d => (x, y, d) (d = gcd a b) gcdext a b = (x, y, d) where d = gcd a b (x, y) = case GMP.gcdExtInteger (fi a) (fi b) of (# p, q #) -> (fI p, fI q) {-# INLINE gcdext #-} modInv :: Int -> Int -> Int modInv a mo | 1 == g = mkPos i | otherwise = -1 where (i, _, g) = gcdext a mo mkPos x | x < 0 = x + mo | otherwise = x {-# INLINE modInv #-} stream :: Monad m => Int -> Int -> VFSM.Stream m Int stream !l !r = VFSM.Stream step l where step x | x < r = return $ VFSM.Yield x (x + 1) | otherwise = return $ VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] stream #-} rep :: Monad m => Int -> (Int -> m ()) -> m () rep n = flip VFSM.mapM_ (stream 0 n) {-# INLINE rep #-} streamR :: (Monad m) => Int -> Int -> VFSM.Stream m Int streamR !l !r = VFSM.Stream step (r - 1) where step x | x >= l = return $ VFSM.Yield x (x - 1) | otherwise = return VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] streamR #-} rev :: (Monad m) => Int -> (Int -> m ()) -> m () rev !n = flip VFSM.mapM_ (streamR 0 n) {-# INLINE rev #-} streamStep :: (Monad m) => Int -> Int -> Int -> VFSM.Stream m Int streamStep !l !r !d = VFSM.Stream step l where step x | x < r = return $ VFSM.Yield x (x + d) | otherwise = return VFSM.Done {-# INLINE [0] step #-} {-# INLINE [1] streamStep #-} infixl 8 `shiftRL`, `unsafeShiftRL` shiftRL :: Int -> Int -> Int shiftRL = unsafeShiftRL {-# INLINE shiftRL #-} unsafeShiftRL :: Int -> Int -> Int unsafeShiftRL (I# x#) (I# i#) = I# (uncheckedIShiftRL# x# i#) {-# INLINE unsafeShiftRL #-} lowerBoundM :: (Monad m) => Int -> Int -> (Int -> m Bool) -> m Int lowerBoundM low high p = go low high where go !low !high | high <= low = return high | otherwise = p mid >>= bool (go (mid + 1) high) (go low mid) where mid = low + unsafeShiftRL (high - low) 1 {-# INLINE lowerBoundM #-} upperBoundM :: (Monad m) => Int -> Int -> (Int -> m Bool) -> m Int upperBoundM low high p = do flg <- p high if flg then return high else subtract 1 <$!> lowerBoundM low high (fmap not.p) {-# INLINE upperBoundM #-} lowerBound :: Int -> Int -> (Int -> Bool) -> Int lowerBound low high p = runIdentity (lowerBoundM low high (return . p)) {-# INLINE lowerBound #-} upperBound :: Int -> Int -> (Int -> Bool) -> Int upperBound low high p = runIdentity (upperBoundM low high (return . p)) {-# INLINE upperBound #-} (.>>.) :: Bits i => i -> Int -> i (.>>.) = unsafeShiftR {-# INLINE (.>>.) #-} (.<<.) :: Bits i => i -> Int -> i (.<<.) = unsafeShiftL {-# INLINE (.<<.) #-} (.>>>.) :: Int -> Int -> Int (.>>>.) = unsafeShiftRL {-# INLINE (.>>>.) #-} ------------------------------------------------------------------------------- type Parser a = BSC8.ByteString -> Maybe (a, BSC8.ByteString) parseInt :: Parser Int parseInt = fmap (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 parseANBN :: Int -> IO (VU.Vector Int, VU.Vector Int) parseANBN n = do vectup <- VU.replicateM n $ (\vec -> (vec VU.! 0, vec VU.! 1)) . VU.unfoldr (BSC8.readInt . BSC8.dropWhile isSpace) <$> BSC8.getLine return $ VU.unzip vectup parseANBNCN :: Int -> IO (VU.Vector Int, VU.Vector Int, VU.Vector Int) parseANBNCN n = do vectup <- VU.replicateM n $ (\vec -> (vec VU.! 0, vec VU.! 1, vec VU.! 2)) . VU.unfoldr (BSC8.readInt . BSC8.dropWhile isSpace) <$> BSC8.getLine return $ VU.unzip3 vectup ------------------------------------------------------------------------------- main :: IO () main = do n <- parse1 rep n $ \_ -> do p <- parse1 if p == 2 then print 2 else print (p * (p - 2) + 1)