LintCode 快乐数

题目描述：
写一个算法来判断一个数是不是”快乐数”。

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

您在真实的面试中是否遇到过这个题？ Yes
样例
19 就是一个快乐数。

1^2 + 9^2 = 82
8^2 + 2^2 = 68
6^2 + 8^2 = 100
1^2 + 0^2 + 0^2 = 1

ac代码：

class Solution {
public:
    /**
     * @param n an integer
     * @return true if this is a happy number or false
     */
    int solve(int n)
    {
        int ans=0;
        while(n)
        {
            int temp=n%10;
            ans+=temp*temp;
            //cout<<temp<<endl;
            n=n/10;
        }
        return ans;
    }
    bool isHappy(int n) {
        // Write your code here
        int flag=n;
        int t=20;
        while(t--)
        {
            flag=solve(flag);
            if(flag==1)
                break;
            //cout<<flag<<endl;
        }
        if(flag==1)
            return 1;
        return 0;
    }
};