LeetCode Happy Number 开心数字

 

题意：

　　给出一个整数n，判断其是否为幸运数。

　　规则是，将n按十进制逐位拆出来后，每个位各自进行取平方，再将这些平方数求和作为新的数字n。若最后n=1，就是幸运数。

 

思路：

　　计算例子：n=47，接着n=4*4+7*7=65，接着n=6*6+5*5=61，接着....

　　注意有可能陷入无限循环，就是迭代的过程产生了一个环路，即n又重复出现了。

class Solution {
public:
    bool isHappy(int n) {
        unordered_set<int> sett;
        while(n!=1)
        {
            if(sett.find(n)!=sett.end())
                return false;
            sett.insert(n);
            int tmp=0;
            while(n)
            {
                tmp+=((n%10)*(n%10));
                n/=10;
            }
            n=tmp;
        }
        return true;
    }
};

 

 

 

python3

class Solution(object):
    def isHappy(self, n):
        """
        :type n: int
        :rtype: bool
        """
        sett=set([n])
        while n>1:
            n=sum([int(x)**2 for x in str(n)])
            if n in sett:    return False
            sett.add(n)
        return True