leetcode -- 202.快乐数

题源

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 <= n <= 231 - 1

思考

思考一

第一点，首先有一个函数能够将一个数替换为它的各位数上的平方和。

第二点，需要一个方法去判断是否重复，这里

我选择用到unordered_set，用while循环判断计

算出来的数是否为1，如果为1，跳出循环，否

则判断此数是出现过，如果出现过，则说明此

数不是一个快乐数，如果没出现过则将其加入

到set中

实现思考一代码

class Solution {
public:

    //通过sum()函数计算一个正整数的平方和。
    int sum(int x){
        int sum = 0;
        while (x > 0){
            sum += pow(x%10,2);
            x /= 10;
        }
        return sum;
    }
    bool isHappy(int n) {
        //定义一个unordered_set来判断n是否无限循环 
        unordered_set<int> res;
        //计算第一次的平方和
        int temp = sum(n);
        while(temp != 1){
            //可能有两种情况是可以断定n不是快乐数
            //第一种就是出现循环
            //但是由于题目提示了n的范围不包括0，所以无需考虑n = 0造成的循环情况
            if (res.find(temp) != res.end()) return false;
            else{
                //temp从未出现过，将其存入res中
                res.insert(temp);
                //还未结束，继续更新平方和
                temp = sum(temp);
            } 
        }
        //快乐数到0，就可以退出循环
        return true;
    }
};

时间复杂度分析

对于sum()函数而言，时间复杂度是x的位数，即O(常数)

对于while循环仍然是个常数级的时间复杂度，它取决于n到1的步数，但即

使n到不了1，进入了循环，时间复杂度仍然是常数级

综上，这个判断一个数是否为快乐数的算法的时间复杂度仍然为O（1）；

注：诸位站友如有所收获不妨点个免费的赞，如有错误之处或有其它补充的点，请在评论区发表你的观点，看到必回。