move askNumber to utils

This commit is contained in:
2025-10-17 15:50:59 +02:00
parent b4146cd8ce
commit 4fc1af5c88
3 changed files with 19 additions and 24 deletions

View File

@@ -1,10 +1,10 @@
import Text.Read (readMaybe)
import System.Exit (exitSuccess)
import Utils (askNumber)
main :: IO ()
main = do
n <- askNumber
n <- askNumber "Enter an integer:"
let (d, root) = findIntegerSqrt n
q = (root - d)
p = (root + d)
@@ -12,19 +12,8 @@ main = do
putStrLn ("Found d = " ++ show d ++ ", sqrt(n + d^2) = " ++ show root)
putStrLn ("q = " ++ show q ++ ", p = " ++ show p)
askNumber :: IO Int
askNumber = do
putStrLn "Enter an integer:"
input <- getLine
case readMaybe input :: Maybe Int of
Just n -> return n
Nothing -> do
putStrLn "Not a valid integer!"
exitSuccess
-- Find the smallest integer d >= 0 such that sqrt(n + d^2) is an integer
findIntegerSqrt :: Int -> (Int, Int)
findIntegerSqrt :: Integer -> (Integer, Integer)
findIntegerSqrt n = go 0
where
go d =