LeetCode 202. Happy Number (快樂數)

題目

連結

https://leetcode.cn/problems/happy-number/

問題描述

編寫一個演算法來判斷一個數 n 是不是快樂數。

「快樂數」 定義為：

對於一個正整數，每一次將該數替換為它每個位置上的數字的平方和。
然後重複這個過程直到這個數變為 1，也可能是 無限迴圈 但始終變不到 1。
如果這個過程 結果為 1，那麼這個數就是快樂數。
如果 n 是 快樂數 就返回 true ；不是，則返回 false 。

示例

輸入：n = 19
輸出：true
解釋：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

提示

1 <= n <= 231 - 1

思路

可以採用雜湊來儲存，或者快慢指標。無限迴圈的話，可以當做一個環，如果不是環，那麼就收斂於1.對於求是否存在環，用快慢指標，一個走兩步，一個走一步，相遇或到1為止。

複雜度分析

時間複雜度 O(n)
空間複雜度 O(1)

程式碼

Java

    public boolean isHappy(int n) {
        int slow = n;
        int fast = get(n);
        while (fast != 1 && fast != slow) {
            fast = get(get(fast));
            slow = get(slow);
        }
        if (fast == 1) {
            return true;
        } else {
            return false;
        }
    }

    public int get(int n) {
        int sum = 0;
        while (n > 9) {
            int d = n % 10;
            n = n / 10;
            sum += d * d;
        }
        sum += n * n;
        return sum;
    }