leetcode刷题（58）—— 202. 快乐数

一、题目

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

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

如果 n 是快乐数就返回 True ；不是，则返回 False 。

示例：

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

二、思路及代码实现

按照规则进行计算时，会有三种可能：

• 最终会得到 1。
• 最终会进入循环。
• 值会越来越大，最后接近无穷大。

进一步分析，

• 两位数最大为 99，而 $9^2+9^2=162$；三位数最大为 999，而 $9^2+9^2+9^2=243$。也就是说，三位数以下的数字，计算的结果总是会小于 243 的。
• 四位数最大为 9999，有 $9^2+9^2+9^2+9^2=324$；……；对于最大的十三位数 9999999999999，有 $9^2+9^2+...+9^2+9^2=1053$。也就是说，四位数以上的数字，计算的结果最终都会回到三位数。
• 由此可见，该过程不会趋于无穷大。要么进入循环，要么最终得到 1。

所以我们的可以通过检查是否有循环来判断是否是快乐数。

思路一：用 HashSet 检测循环

按照规则每次计算一个平方和，

• 如果它不在 HashSet 中，将其添加至 HashSet ；
• 如果它在 HashSet 中，这意味着我们处于一个循环中，因此应该返回 false。

参考代码：

class Solution {
    public boolean isHappy(int n) {
        Set<Integer> set = new HashSet<>();
        while(n != 1 && !set.contains(n)){
            set.add(n);
            n = getNext(n);
        }
        return n == 1;
    }

    public int squareSum(int n){
        int sum = 0;
        while(n != 0){
            int d = n % 10;
            sum += d * d;
            n = n / 10;
        }
        return sum;
    }
}

思路二：用 快慢指针 检测循环

快指针每次走两步，慢指针每次走一步，如果有环，两个指针一定会相遇。

参考代码：

class Solution {
    public boolean isHappy(int n) {
        int slow = n;
        int fast = squareSum(n);
        while(slow != fast){
            slow = squareSum(slow);
            fast = squareSum(squareSum(fast));
        }
        return slow == 1;
    }

    public int squareSum(int n){
        int sum = 0;
        while(n != 0){
            int d = n % 10;
            sum += d * d;
            n = n / 10;
        }
        return sum;
    }
}