LeetCode 202. Happy Number

happy number的定义：不断的将每一位平方并相加，最后能得到1的数字。

核心思想：如果不是happy number的话，那么这个数会陷入死循环，即每位平方和的结果会和之前的结果重复。略微带点动态规划的思想（？）

class Solution {
public:
    bool isHappy(int n) {
        set<int> processed;
        while(n != 1 && processed.insert(n).second){
            int sum = 0;
            while(n){
                sum += (n % 10) * (n % 10);
                n = n / 10;
            }
            n = sum;
        }
        return n == 1;
    }
};

注：set.insert(x)返回的是一个pair，first是一个指向x的迭代器（不论插入前x是否已经在set中），second是bool，如果x已经存在返回false。

 

另外一种算法：

核心思想是4不是happy number，因此不用维护一个set，当数字循环到1或4时即可判断是否happy。

bool isHappy(int n) {
        int num=0;
        while(n!=1&&n!=4)
        {
            while(n)
            {
                num += (n%10) * (n%10);
                n/=10;
            }
            n=num;num=0;
        }
        return 1==n;
    }

两种算法时间相同。