LintCode-488.快乐数

题目

描述

写一个算法来判断一个数是不是"快乐数"。

一个数是不是快乐是这么定义的：对于一个正整数，每一次将该数替换为他每个位置上的数字的平方和，然后重复这个过程直到这个数变为1，或是无限循环但始终变不到1。如果可以变为1，那么这个数就是快乐数。

样例

19 就是一个快乐数。

1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

解答

思路

1. 计算过程中的数字都存在Set中，如果后续计算出现了Set中的数，说明陷入循环，那么该Set中的所有数都应该不是快乐数，当然也包括输入的数，返回false；
2. 直到计算出1，返回true；

代码

public class Solution {
    /**
     * @param n an integer
     * @return true if this is a happy number or false
     */
    public boolean isHappy(int n) {
        // Write your code here
        Set set = new HashSet<>();
        int temp = 0;
        Integer nn = new Integer(n);
        while(nn != 1){
            if(set.contains(nn)) return false;
            set.add(nn);
            temp = nn;
            nn = 0;
            while(temp != 0){
                nn += (temp%10)*(temp%10);
                temp = temp / 10;
            }
        }
        return true;
    }
}