LeetCode-202. 快乐数

题目

编写一个算法来判断一个数是不是“快乐数”。

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

示例:

输入: 19
输出: true
解释: 
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

解题

• 使用Set来解答
• 计算过程中如果结果不为1, 则将其放入Set中
• 放入之前进行判断, 如果Set中已经存在了, 说明已经形成了一个循环, 就一定不是快乐数了
  , 代码如下:

class Solution {
    public boolean isHappy(int n) {
        Set<Integer> set = new HashSet<>();
        set.add(n);
        while (true){
            n = getSquaresum(n);
            if (set.contains(n)){
                if (n == 1){
                    return true;
                } else {
                    return false;
                }
            } else {
                set.add(n);
            }
        }
    }

    /**
     * 获取一个数字各个位上的平方和
     * @param num
     * @return
     */
    private int getSquaresum(int num){
        int sum = 0;
        int temp;
        while (num != 0){
            temp = num % 10;
            sum += temp * temp;
            num = num / 10;
        }
        return sum;
    }
}