結果
問題 | No.1235 ζ関数 |
ユーザー | こまる |
提出日時 | 2020-10-02 13:12:42 |
言語 | Haskell (9.8.2) |
結果 |
AC
|
実行時間 | 4 ms / 2,000 ms |
コード長 | 1,256 bytes |
コンパイル時間 | 7,731 ms |
コンパイル使用メモリ | 189,980 KB |
実行使用メモリ | 7,924 KB |
最終ジャッジ日時 | 2023-09-22 15:32:30 |
合計ジャッジ時間 | 8,802 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 3 ms
7,728 KB |
testcase_01 | AC | 3 ms
7,860 KB |
testcase_02 | AC | 3 ms
7,924 KB |
testcase_03 | AC | 3 ms
7,896 KB |
testcase_04 | AC | 3 ms
7,768 KB |
testcase_05 | AC | 3 ms
7,824 KB |
testcase_06 | AC | 3 ms
7,816 KB |
testcase_07 | AC | 3 ms
7,848 KB |
testcase_08 | AC | 3 ms
7,820 KB |
testcase_09 | AC | 3 ms
7,764 KB |
testcase_10 | AC | 3 ms
7,872 KB |
testcase_11 | AC | 3 ms
7,756 KB |
testcase_12 | AC | 3 ms
7,868 KB |
testcase_13 | AC | 3 ms
7,796 KB |
testcase_14 | AC | 4 ms
7,816 KB |
testcase_15 | AC | 3 ms
7,836 KB |
testcase_16 | AC | 4 ms
7,840 KB |
testcase_17 | AC | 3 ms
7,780 KB |
testcase_18 | AC | 3 ms
7,820 KB |
testcase_19 | AC | 3 ms
7,824 KB |
testcase_20 | AC | 3 ms
7,844 KB |
コンパイルメッセージ
Loaded package environment from /home/judge/.ghc/x86_64-linux-9.6.1/environments/default [1 of 2] Compiling Main ( Main.hs, Main.o ) [2 of 2] Linking a.out
ソースコード
import Control.Monad import Control.Monad.ST import Data.Bits import qualified Data.Vector as V import qualified Data.Vector.Mutable as VM import qualified Data.Vector.Unboxed as VU import qualified Data.Vector.Unboxed.Mutable as VUM main :: IO () main = readLn >>= zeta >>= print zeta :: Int -> IO Double zeta n = do let primes = getPrimeVector 10000 xm <- V.unsafeThaw $ V.fromList [1.0] forM_ [0 .. (VU.length primes - 1)] $ \i -> do let p = fromIntegral $ primes VU.! i x = 1.0 / (1.0 - p ** (- (fromIntegral n))) VM.modify xm (\a -> a * x) 0 (V.! 0) <$> V.unsafeFreeze xm getPrimeVector :: Int -> VU.Vector Int getPrimeVector top = VU.filter (/= -1) . VU.imap (\i check -> if i == 0 then 2 else if check then i * 2 + 1 else -1) $! runST $ do let m = (top - 1) `shiftR` 1 r = floor . sqrt . fromIntegral $ (top + 1) sieve <- VU.unsafeThaw $ VU.replicate (m + 1) True forM_ [1 .. r `unsafeShiftR` 1] $ \i -> do isPrime <- VUM.unsafeRead sieve i when isPrime $ do forM_ [2 * i * (i + 1), 2 * i * (i + 2) + 1 .. m] $ \j -> do VUM.unsafeWrite sieve j False VU.unsafeFreeze sieve