202

202.快乐数

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

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

示例:

输入: 19
输出: true
解释: 
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1

暴力计算

bool isHappy(int n) {
    int sum = 0;
    while (true) {
        if ((n == 2 || n == 3 || n == 4) && sum == 0) return false;
        sum += (n % 10) * (n % 10);
        n /= 10;
        if (n == 0 && sum == 1) {
            return true;
        } else if (n == 0 && sum != 1) {
            n = sum;
            sum = 0;
        }
    }
}

思维优化

int getNext(int x) {
    int ans = 0;
    while (x) {
        ans += (x % 10) * (x % 10);
        x /= 10;
    }
    return ans;
}

bool isHappy(int n) {
    int p = n, q = n;
    do {
        p = getNext(p);
        q = getNext(getNext(q));
    } while(p - q && q - 1);
    return q == 1;