“玲珑杯”Round #11

D -- 萌萌哒的第四题

Time Limit：2s Memory Limit：128MByte

Submissions：274Solved：182

DESCRIPTION

给一个数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

#include <bits/stdc++.h>
using namespace std;

int cnt;
void f(int x)
{
	cnt++;
	int ans=0;
	while(x>0)
	{
		ans+=(x%10)*(x%10);
		x/=10;
	}
	if(ans==1)
	{
		cout<<"YES"<<endl;
		return ;
	}
	if(cnt>=100)		//如果超过100次还没有找到，就默认不是‘特别的数’ 
	{
		cout<<"NO"<<endl;
		return ;
	}
	else
	{
		f(ans);
	}
}
int main()
{
	int a;
	while(~scanf("%d",&a))
	{
		cnt=0;	//每次都要初始化为0 
		f(a);
	}
	return 0;
}