题目：
一个数字链由数字的每位平方相加而成。
例子：44->32->13->10->1->1 .4*4+4*4=32,3*3+2*2=13 .........
给你一个数字N计算出N以内的所有符合要求的数的个数（包括N）。1<=N<=5000000
输入包含一个T，表示测试次数。接着是一个整数N
输入样例：
4
1
10
1000
10000
输出样例：
1
3
143
1442
我的代码（错误的- -）：
#include <iostream>
#include <vector>
using namespace std;
vector <int> cyj; // 保存符合要求的数字。
vector <int> tt; // 临时数组。
int a[5000001]; // 用于标记符合要求的数字1表示是，0表示不是。
void judge(int n)
{
tt.insert(tt.begin(),n); //先将n插入tt数组。
if(n==1||a[n]==1) // 如果n==1或者n为已经标记的数字则执行一下语句。
{
for(int j=0;j<tt.size();j++) // 遍历tt数组
{
if(a[tt[j]]==0) // 如果tt[j]的值未被标记则
{ // 标记该数字，并且将值插入cyj数组。
a[tt[j]]=1;
cyj.insert(cyj.begin(),tt[j]);
}
}
tt.clear(); // 清空tt数组，返回。
return ;
}
int m=n; // 用m 保留n的原值
int sum=0;
while(n>0) // 将n每位数字提取并把平方的值与sum相加。
{
int t=n/10;
sum+=t*t;
n/=10;
}
if(sum>m) // 如果sum大于n原来的值则该数不符合要求直接返回。
{
return ;
tt.clear();
}
else // 否则
{
judge(sum);
}
if(sum<10&&sum!=1) // 如果sum小于10且不等于1则返回。
{
tt.clear();
return ;
}
}
int main()
{
int T;
cin>>T;
while(T--) // T为测试次数
{
memset(a,0,sizeof(a)); // a数组初始化为0
int n;
cin>>n;
if(n==1) //1，和小于10的情况单独考虑。
{
cout<<'1'<<endl;
continue;
}
else if(n<10)
{
cout<<'2'<<endl;
continue;
}
a[1]=1; // 1,10是符合要求的数字 。
a[10]=1;
cyj.insert(cyj.begin(),1); // 因为题目规定有两个1所以cyj里面插入两个1.
cyj.insert(cyj.begin(),1);
for(int i=10;i<=n;i++)
{
if(a[i]==1) continue; // 如果i是已经标记的数则继续循环。
judge(i);
}
cout<<cyj.size()<<endl;
}
return 0;
}
