hdu 1999 不可摸数

#include <iostream>
#include <cstdio>
#include <cmath>

using namespace std;

int a[1005];
int b[1000005];

void not_toch()
{
memset(a,0,sizeof(a));
for(int i=0;i<1000005; i++)b[i] = 1;

for(int i=2;i<=500005;i++)
{
for(int j=i+i;j<1000005;j+=i)
b[j] += i;

}
for(int i=1;i<1000001;i++)
if(b[i]<1001) a[b[i]] = 1;
}
int main()
{
int test;
int n;
not_toch();
cin>>test;
while(test--)
{
cin>>n;
if(a
) cout<<"no"<<endl;
else cout<<"yes"<<endl;
}
return 0;
}