快乐数

package main
//import "fmt"

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param n int整型 
 * @return bool布尔型
*/
func happynum( n int ) bool {
    //如果不能变成1，就会陷入循环
    cnt:=map[int]bool{}
    for n!=1{
        if _,ok:=cnt[n];ok{
            return false
        }
        cnt[n]=true
        n=handle(n)
    }
    return true
}

func handle(n int)int{
    ans:=0
    for n>0{
        ans+=(n%10)*(n%10)
        n/=10
    }
    return ans
}