快樂數

柴勇玛喜达發表於2024-09-06

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

「快樂數」定義為：

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

如果n是快樂數就返回true;不是，則返回false。

示例 1：

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

示例 2：

輸入：n = 2
輸出：false

class Solution {
    private int getNext(int n) {
        int totalSum = 0;
        while (n > 0) {
            int d = n % 10;
            n = n / 10;
            totalSum += d * d;
        }
        return totalSum;
    }

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

相關文章