92.hs

import List ( elemIndex )
import Char ( digitToInt )

check ones nines [] = length nines
check ones nines (x:xs) = if (ending x) == 1 then check (x:ones) nines xs else 1 + check ones (x:nines) xs
    where
        ending 1 = 1
        ending 89 = 89
        ending x  = case elemIndex x ones of
                         Just _  -> 1
                         Nothing -> case elemIndex x nines of
                                        Just _  -> 89
                                        Nothing -> ending' x
        ending' x = ending $ (sum $ map (^2) $ map digitToInt $ show x)
    

checkCount xs = check [] [] xs

main = print $ checkCount [1..10000000]