getLine x次haskell

Question

基于Hackerrank的问题

我的问题是如何在stdIn上进行getLine t次？

main = do
    t <- getInt
    let x = [divisorsInNumber unsafeGetInt | a <-[1..t] ]
    print x

getInt :: IO Int
getInt = fmap read getLine

unsafeGetInt :: Int
unsafeGetInt = unsafePerformIO getInt

divisorsInNumber n = length $ filter (== True) $ map (isDivisor n) (integralToListOfInts n)

Answer 1

只是复制t次的getLine与操作replicateM ：

import Control.Monad (replicateM)

getLines :: Int -> IO [String]
getLines t = replicateM t getLine

因此getInts ，即getInt t次，可以表示为：

getInts :: Int -> IO [Int]
getInts = fmap read <$> getLines

重写为使用getInts的完整代码可以是：

import Control.Applicative ((<$>))
import Control.Monad (replicateM)


getLines :: Int -> IO [String]
getLines n = replicateM n getLine

getInts :: Int -> IO [Int]
getInts n = fmap read <$> getLines n

getInt :: IO Int
getInt = fmap read getLine

divisorsInNumber :: Int -> Int
divisorsInNumber n = length $ filter (isDivisor n) (integralToListOfInts n)

main :: IO ()
main = do
    t <- getInt
    nums <- getInts t
    let x = [divisorsInNumber num | num <- nums]
    print x