玲珑杯 1099萌萌哒的第四题

给一个数x，定义一个函数f(x)的结果是x的各位数字的平方和，若经过无数次递归操作之后若结果为1，也就是f(f(f(...f(x)...)))=1，那么这个数被认为是一个特别的数。给出一个数x请问这个数是否特别。

INPUT

包含多组测试数据(<=20)，每组数据一行一个整数x(1<=x<=1000000000)

OUTPUT

每组数据输出一行YES表示是一个特别的数，否则输出NO

SAMPLE INPUT

19

2

14

SAMPLE OUTPUT

YES

NO

NO 

我是记录了递归的次数，感觉递归了100次都不是1的话，肯定就不是了

#include<iostream>
#include<functional>
#include<queue>
#include<cstring>
#include<cstdio>
using namespace std;
char a[20];
void F(char a[],int ans){
   if(ans>100)
  {
   printf("NO\n");
   return ;
  }
int i;
char b[20];
long long sum=0;
int l=strlen(a);
for(i = 0; i < l;i ++){
    sum += (a[i]-'0')*(a[i]-'0');
}
    if(sum == 1){
        printf("YES\n");
        return ;
    }
    else {
            i=0;
        memset(b,0,sizeof(b));
      while(sum!=0){
        b[i]=sum%10+'0';
         sum/=10;
         i++;
      }
       F(b,ans+1);
    }
}
int main(){
 while(scanf("%s",a)!=EOF){
    F(a,1);
 }
}