LeetCode202 快乐数

题目

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

示例

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

C++代码

首先将数字n逐位存入vector容器中（方便计算平方和），然后将其平方和与1进行比较，相同则返回true，不同则继续。
但是注意，非快乐数永远不会出现平方和等于1的情况，所以如果不设置停止条件，会死循环。可以设置为，当得到的平方和已经出现过，则说明它不是快乐数，返回false。

class Solution {
public:
    bool isHappy(int n) 
    {
        vector<int> nums;
        vector<int> save;
        getnum(n,nums);
        int sum=calculate(nums);
        while(sum!=1) //平方和等于1，跳出循环，返回true
        {
     	    //若平方和已经出现过，则返回false
            if(find(save.begin(),save.end(),sum)!=save.end()) return false;
            else
            {
                save.push_back(sum); //未出现，则保存下来
                getnum(sum,nums);
                sum=calculate(nums);                
            }
        }
        return true;
        
        
    }
    void getnum(int n,vector<int> &num) //逐位存储到vector容器中（逆序）
    {
        num.clear();
        while(n)
        {
            num.push_back(n%10);
            n/=10;
        }
    }
    int calculate(vector<int> num) //计算平方和
    {
        int sum=0;
        for(vector<int>::iterator iter=num.begin();iter!=num.end();iter++)
            sum=sum+(*iter)*(*iter);
        return sum;
    }
};