Write a script to find "Happy Numbers"

These challenge posts have been a lot of fun. Another one inspired by Fermat's Library.

Challenge

In the language of your choice, write a script to find "Happy Numbers." Via Wikipedia:

A happy number is a number defined by the following process: Starting with any positive integer, replace the number by the sum of the squares of its digits in base-ten, and repeat the process until the number either equals 1 (where it will stay), or it loops endlessly in a cycle that does not include 1. Those numbers for which this process ends in 1 are happy numbers, while those that do not end in 1 are unhappy numbers (or sad numbers).

Looking at the picture in the embedded tweet will probably help solidify the process. Here are the first few happy numbers to help check your work:

1, 7, 10, 13, 19, 23, 28, 31

Have fun!

Comments

[Vinay Pai]

Here's a golang implementation, optimized for speed. It takes about 200ms on my machine to find all the happy numbers from 1 to a million on my machine if you take out the prints.

package main

import (
    "fmt"
)

func sumDigitsSquare(n int) int {
    sum := 0
    for n != 0 {
        digit := n % 10
        sum += digit * digit
        n /= 10
    }
    return sum
}

func isHappy(n int, memo map[int]bool) bool {
    if result, seen := memo[n]; !seen {
        result = isHappy(sumDigitsSquare(n), memo)
        memo[n] = result
    }
    return memo[n]
}

func main() {
    memo := make(map[int]bool, 1000000)
    memo[1] = true
    memo[4] = false

    for i := 1; i <= 1000000; i++ {
        if isHappy(i, memo) {
            fmt.Println(i)
        }
    }
}

[Colin Kiama]

 private static string FindHappyNumbers()
        {
            Console.WriteLine("How many happy numbers do you want to find?");


            int NumbersToFind = int.Parse(Console.ReadLine());
            int[] NumbersFound = new int[NumbersToFind];
            int HappyNumbersFound = 0;
            int startingNum = 1;
            int num = 1;
            const int searchLoopLimit = 10;
            int searchLoops = 0;
            string numAsString = String.Empty;
            bool searchEnded = false;

            while (HappyNumbersFound < NumbersToFind)
            {
                if (searchLoops == searchLoopLimit)
                {
                    searchEnded = true;
                }

                if (searchEnded)
                {
                    ResetLoopWithNextNumber(ref startingNum, ref num, ref searchLoops, ref searchEnded);

                }

                numAsString = num.ToString();
                int sum = 0;
                foreach (var digitChar in numAsString)
                {
                    int digit = int.Parse(digitChar.ToString());
                    sum += (int)Math.Pow(digit, 2);
                }

                if (sum == 1)
                {
                    NumbersFound[HappyNumbersFound] = startingNum;
                    HappyNumbersFound += 1;
                    searchEnded = true;
                }
                else
                {
                    num = sum;
                    searchLoops += 1;
                }
            }

            string FormattedArray = TurnArrayIntoWellFormattedString(NumbersFound);
            return FormattedArray;

        }

 private static void ResetLoopWithNextNumber(ref int startingNum, ref int num, ref int searchLoops, ref bool searchEnded)
        {
            startingNum += 1;
            num = startingNum;
            searchLoops = 0;
            searchEnded = false;
        }

        private static string TurnArrayIntoWellFormattedString(int[] numbersFound)
        {
            string StringToReturn = "";
            int length = numbersFound.Length;

            for (int i = 0; i < length; i++)
            {
                if (i < length - 1)
                {
                    StringToReturn += numbersFound[i] + ", ";
                }
                else
                {
                    StringToReturn += numbersFound[i];
                }
            }

            return StringToReturn;
        }

Here's mine in C#. I feel like this could have been way shorter though. Oh well :p

[NAVID MANSURI]

bro log code is not good......

[Colin Kiama]

Thanks for letting me know.

What should I change so that my code improves?

[NAVID MANSURI]

your feedback...
your code write is good and logic is also good , but you code very long ,,,

so, listen ,try to write code small, simpal

[Vikas Uikey]

Golang Solution

package main

import (
    "fmt"
    "strconv"
    "strings"
)

func isHappyNumber(num int) bool {
    newNum := 0
    switch num {
    case 1:
        return true
    case 4:
        return false
    default:
        s := strconv.Itoa(num)
        digits := strings.Split(s, "")
        for _, i := range digits {
            n, _ := strconv.Atoi(i)
            newNum += n * n
        }
    }
    return isHappyNumber(newNum)
}

func main() {
    fmt.Println(isHappyNumber(94))
}

[Sebastien Castiel]

Can’t resist, here’s a solution in Reason: (Try it)

let rec digits = (n) => n === 0 ? [] : [n mod 10, ...digits(n / 10)];

let squaresSum = (digits) =>
  digits |> List.map((n) => n * n) |> List.fold_left((a, b) => a + b, 0);

let rec isHappy = (n) =>
  switch n {
  | 1 => true
  | 4 => false
  | _ => isHappy(n |> digits |> squaresSum)
  };

let rec nextHappyNumber = (n) => isHappy(n) ? n : nextHappyNumber(n + 1);

let firstHappyNumbers = (nb) => {
  let rec firstHappyNumbersFrom = (nb, found) =>
    if (nb === 0) {
      found
    } else {
      let from = List.length(found) === 0 ? 1 : List.hd(found) + 1;
      let next = nextHappyNumber(from);
      firstHappyNumbersFrom(nb - 1, [next, ...found])
    };
  firstHappyNumbersFrom(nb, [])
};

Js.log(firstHappyNumbers(20));

[Heiko Dudzus]

I'm glad to see some Haskell here. :) I somehow prefer your computational approach to split the number over my quick string hack. You can easily change the base.

I have done a lot of parsing stuff in Haskell with parsec, attoparsec and readP, lately. So I instantly think of parsing when I see the problem of splitting a number into digits. The detour via strings is comfortable, because it uses the read parser. but

"Challenge" accepted to solve it computationally. :)

To import as few / basic libraries as reasonably achievable, I decided not to use the parsec library but to write a homemade digit parser.

What you do in splitNum reminds me of the quotRem function (that I found when I coded for the Challenge about Kaprekar numbers, here). quotRem just needs a swap to make for a nice digit parser in a very organic way: When dividing by 10 the integer quotient is the state of the state monad (that's still to parse) and the remainder is a parsed digit.

import Control.Monad.State
import Control.Monad.Loops(untilM)

swap :: (a,b) -> (b,a)
swap (x,y) = (y,x)

-- digit parser
digit :: Integral a => State a a
digit = state $ \s -> swap $ quotRem s 10

-- eof condition
eof :: Integral a => State a Bool
eof = get >>= \s -> return $ s == 0

-- parsing the digits of a number into a (reversed) list of digits
digits :: Integral a => a -> [a]
digits = evalState (digit `untilM` eof)

-- core function
f = sum . map (^2) . digits

-- iteration of f by recursion of isHappy 
isHappy n | n == 4 = False
          | n == 1 = True
          | otherwise = isHappy (f n)

I am a bit biased to use (faster) Ints in my prototypes, and use Integers where really needed, but I adopt your approach with Integers by at least generalizing the code to Integral types.

Although I learned to prefer folds over homemade recursion, I'm afraid the abstractness of foldP is a little bit hard to read, so perhaps I stick with the homemade recursion, this time. :)

[Bohdan Stupak]

F#

let rec extractDigits acc input = 
    if input <= 9 then input::acc  
    else extractDigits (input % 10::acc) (input/10)

let sumOfSquares input = 
    input
    |> List.fold(fun acc i -> acc + i*i) 0

let rec isHappy input =
    if input = 1 then true
    else if input >= 2 && input <=6 then false
    else 
        let newInput = 
            input
            |> extractDigits List.empty
            |> sumOfSquares
        isHappy newInput

[<EntryPoint>]
let main argv =
    [1..10000]
    |> List.filter(fun i -> isHappy i)
    |> List.iter(fun i -> printfn "%d" i)        
    0

[Dave Jacoby]

A work in two parts. First, Happy.pm, a Perl5 Module.

package Happy ;

use strict ;
use warnings ;
use utf8 ;
use feature qw{ postderef say signatures state } ;
no warnings qw{ experimental::postderef experimental::signatures } ;

use Carp ;
use Exporter qw( import ) ;
use List::Util qw( sum0 ) ;

our @EXPORT = qw( is_happy ) ;
our %EXPORT_TAGS = ( 'all' => [ @EXPORT ], ) ;
our @EXPORT_OK = ( @{ $EXPORT_TAGS{ 'all' } } ) ;
our $VERSION = 0.0.1 ;

sub is_happy ( $number ) {
    croak 'Not a number' if $number =~ m{\D} ;
    croak 'Not anything' if !defined $number ;
    my %done ;
    while ( $number != 0 && $number != 1 ) {
        return 0 if $done{ $number } ;
        my $output = sum0 map { $_ ** 2 } split m{}, $number ;
        $done{ $number } = $output ;
        $number = $output ;
        }
    return $number ;
    }
1 ;

Which is called by a program.

#!/usr/bin/env perl

use strict ;
use warnings ;
use utf8 ;
use feature qw{ say } ;

use lib '/home/jacoby/lib' ;
use Happy ;

for my $i ( 0 .. 1_000_000 ) {
    say $i if is_happy( $i ) ;
    }

Not particularly golfy, because I like reading and understanding things.

[Jan]

Here is my C# implementation:

private static bool IsHappy(int n)
{
    var u = n
        .ToString()
        .Select(c => int.Parse(c.ToString()))
        .Sum(d => d * d);

    if (u == 1) return true;
    if (u == 4) return false;
    return IsHappy(u);
}

[Michael Neu]

The obvious pythonic way:

from happy import find_happy_numbers

if __name__ == "__main__":
    for number in find_happy_numbers(100):
        print(number)

Jokes aside, here's a Python implementation using simple memoization:

def memoize(fn):
    cache = {}

    def memoized_fn(*args):
        if args in cache.keys():
            return cache[args]

        cache[args] = apply(fn, args)

        return cache[args]

    return memoized_fn


@memoize
def is_happy(n):
    next = sum([int(i)**2 for i in str(n)])

    if next == 1:
        return True
    elif next > 9:
        return is_happy(next)
    else:
        return False


def find_happy_numbers(n):
    for i in range(n):
        if is_happy(i):
            yield i

[Jay]

A simple Rust Solution

fn is_happy(mut n :i32) -> bool {
    let mut sum = 0;

    while n > 0 {
        let a = n%10;
        n = n/10;
        sum += a*a;
    }

    if sum == 1 {
        return true;
    } else if sum == 4 {
        return false
    }
    is_happy(sum)
}