Created
July 22, 2018 10:03
-
-
Save vilu/754335245789df3b3f95a8f7f3212e0c to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
module Kata.CubeFinder where | |
-- Your task is to construct a building which will be a pile of n cubes. The cube at the bottom will have a volume of n^3, the cube above will have volume of (n-1)^3 and so on until the top which will have a volume of 1^3. | |
-- You are given the total volume m of the building. Being given m can you find the number n of cubes you will have to build? | |
-- The parameter of the function findNb (find_nb, find-nb, findNb) will be an integer m and you have to return the integer n such as n^3 + (n-1)^3 + ... + 1^3 = m if such a n exists or -1 if there is no such n. | |
-- Examples: | |
-- findNb(1071225) --> 45 | |
-- findNb(91716553919377) --> -1 | |
-- Brute force, didn't work | |
-- findNb :: Integer -> Integer | |
-- findNb m = recurse m 0 where | |
-- recurse 0 n = n - 1 | |
-- recurse m n = if (m < 0) | |
-- then - 1 | |
-- else recurse (m - (n ^ 3)) (n + 1) | |
import Debug.Trace | |
findNb :: Integer -> Integer | |
findNb m = if (isInt n) then round n else -1 | |
where | |
n = cuadraticFormula (fromInteger 1) (fromInteger 1) c | |
c = -2 * sqrt (fromInteger m ) | |
isInt x = x == fromInteger (round x) | |
cuadraticFormula :: Double -> Double -> Double -> Double | |
cuadraticFormula a b c = trace (show a ++ " | " ++ show b ++ " | " ++ show c) ((-b) + sqrt (b ^ 2 - (4 * a * c))) / (2 * a) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment