leetcode202. 快乐数

题目描述

编写一个算法来判断一个数 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

分析思路

这道题首先想到的是一个函数,用来求一个数各个位数的平方和。然后在主函数中用一个while(1)循环来重复判断这个和是否等于1。

注意:题目中说无限循环是什么意思呢?就是说,如果这个数不是快乐数,那么在set集合中,一定会出现sum值循环,而且会多次出现。借用评论区的一个回答:
为啥一定不会出现死循环,因为int类型最大值为为‭‭2 147 483 647‬‬, 所以平方和最大的数是1 999 999 999,平方和为1 + 81*9 = 724。任何数的平方和都在1到724之间,724次循环之内一定有重复的。

需要用一个unordered_set来存储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> res;
        // 无限循环
        while(1){
            int sum = getSum(n);
			// 如果等于1,返回true
            if(sum == 1){
                return true;
            }
			// 如果集合中找到了这个值,返回false
            if(res.find(sum)!=res.end()){
                return false;
            }else{
            // 否则插入
                res.insert(sum);
            }
            // 给n赋一个新的值
            n = sum;
        }
    }
};

你可能感兴趣的:(哈希表,力扣刷题,算法)