Leetcode第202题. 快乐数

题目描述

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

「快乐数」 定义为：

• 对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
• 然后重复这个过程直到这个数变为 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

提示：

题目分析

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

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

NOTE1：需要注意的是，每个位置的平方和是一个循环的过程，如果出现1，就返回true，但是会可能无限循环，所以要判断如果出现重复的数就返回false。

NOTE2:  由于要判断这个和，是否重复出现（也就是要快速判断一个元素是否出现在集合中）可以使用哈希法

NOTE3：  数值比较大，且不是（Key，value），选择使用set，且没有顺序要求，使用unordered_set

getSum函数

这个函数的作用是计算整数n的每个位上的数字平方之和

int getSum(int n) {
    int sum = 0; // 初始化sum变量，用于存储各位数字平方的和
    while (n) { // 当n不为0时执行循环
        sum += (n % 10) * (n % 10); // 取n的最低位数字，平方后加到sum上
        n /= 10; // n去掉最低位，准备下一次循环
    }
    return sum; // 返回所有位数的平方和
}

 isHappy函数

 判断整数n是否是一个快乐数。通过不断计算位数平方和，并检查是否重复，来判断是否陷入了非1循环

bool isHappy(int n) {
    unordered_set<int> set; // 使用一个哈希集合来存储出现过的数字
    while(1) { // 开始一个无限循环，直到内部通过break跳出
        int sum = getSum(n); // 调用getSum函数获取当前数字n的位数平方和
        if (sum == 1) { // 如果位数平方和为1
            return true; // n是一个快乐数，返回true
        }
        // 如果这个sum已经在set中（之前出现过这个sum）
        if (set.find(sum) != set.end()) {
            return false; // 已经陷入循环，n不是一个快乐数，返回false
        } else { // 如果sum没有出现过
            set.insert(sum); // 将sum添加到集合中，表示这个值已经计算过
        }
        n = sum; // 更新n为sum，用于下一轮计算
    }
}

完整代码 

class Solution {
public:
    // 取数值各个位上的单数之和
    int getSum(int n) {
        int sum = 0;
        while (n) {
            sum += (n % 10) * (n % 10);
            n /= 10;
        }
        return sum;
    }
    bool isHappy(int n) {
        unordered_set<int> set;
        while(1) {
            int sum = getSum(n);
            if (sum == 1) {
                return true;
            }
            // 如果这个sum曾经出现过，说明已经陷入了无限循环了，立刻return false
            if (set.find(sum) != set.end()) {
                return false;
            } else {
                set.insert(sum);
            }
            n = sum;
        }
    }
};

参考文章：代码随想录 (programmercarl.com)