问题描述
写一个算法来判断一个数是不是”快乐数”。
一个数是不是快乐是这么定义的:对于一个正整数,每一次将该数替换为他每个位置上的数字的平方和,然后重复这个过程直到这个数变为1,或是无限循环但始终变不到1。如果可以变为1,那么这个数就是快乐数。
样例
19 就是一个快乐数。
1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1
求解
class Solution {
public:
/*
* @param n: An integer
* @return: true if this is a happy number or false
*/
bool isHappy(int n) {
unordered_set<int> visited;
auto nextNumber = [](int num) -> int {
int sum = 0;
while(num) {
int rem = num % 10;
sum += rem * rem;
num /= 10;
}
return sum;
};
while(n != 1 && !visited.count(n)) {
visited.emplace(n);
n = nextNumber(n);
}
return n == 1;
}
};