题解 | #奶牛快乐数#

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param n int整型 
     * @return bool布尔型
     */
    int getNewNum(int n) {
        int sum = 0;
        while (n) {
            sum += pow(n%10, 2);
            n /= 10;
        }
        return sum;
    }
    bool isHappy(int n) {
        // write code here
        map<int, int> mp;
        while (n != 1) {
            n = getNewNum(n);
            if (mp[n]) {
                return false;
            } else {
                mp[n] = 1;
            }
        }
        return true;
    }
};