【小米oj】寻找归一数字

骗分过样例，

#define mm(a) memset(a,0,sizeof(a));
#define max(x,y) (x)>(y)?(x):(y)
#define min(x,y) (x)<(y)?(x):(y)
#define Fopen freopen("1.in","r",stdin); freopen("m.out","w",stdout);
#define rep(i,a,b) for(ll i=(a);i<=(b);i++)
#define per(i,b,a) for(ll i=(b);i>=(a);i--)
#include<bits/stdc++.h>
typedef long long ll;
#define PII pair<ll,ll>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=(int)2e6+5;

int n;
bool ck(int n)
{
    int x=n;
    int cnt=0;
    while(x!=1){
        if(cnt>10000)return 0;
        cnt++;
        int now=0;
        while(x){
            now+=(x%10)*(x%10);
            x/=10;
        }
        x=now;
    }
    return 1;
}
int main()
{
    scanf("%d",&n);
    if(ck(n))printf("true
");
    else printf("false
");
    return 0;
}

暴力出奇迹（可以开个vis数组记录然后记忆化搜索，不过懒得写了）