快乐数

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

「快乐数」 定义为：

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

如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

示例 1：

1
2
3
4
5
6
7
输入：n = 19
输出：true
解释：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

示例 2：

1
2
输入：n = 2
输出：false

提示：

  • 1 <= n <= 231 - 1

这道题有两种情况：

  • 是快乐数，经过一系列计算后n=1,返回真
  • 不是快乐数，n会在几个数中循环，就像一个环形链表

第二种情况的证明过程很复杂，用到了不知道是极限还是无穷级数，没仔细看。总之挺复杂。但这道题的标签是简单。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {

    public int getNext(int n){
        int total = 0;
        while(n>0){
            int d = n%10;
            total +=d*d;
            n/=10;
        }
        return total;
    }

    public boolean isHappy(int n) {
        Set<Integer> set = new HashSet<>();
        // 只会有两种情况
        // 1. 是快乐数，n=1 -> return true;
        // 2. 不是快乐数，n会在几个数中循环。
        //
        while(n!=1){
            if(set.contains(n)){
                return false;
            }
            set.add(n);
            n=getNext(n);
        }
        return true;
    }
}