hdoj 1999 不可摸数 【数学】

#include <stdio.h>
#include <string.h>
#define M 500000
int a[M];
int ok[M];
void f()
{
    int i, j;
    for(i = 1; i < M; i ++){
        for(j = 1; j*i < M; j ++){
            a[i*j]+=i;
        }
    }
    for(i = 1; i < M; i ++){
        a[i]-=i;
        if(a[i] < 1001){
            ok[a[i]] = 1;
        }
    }
} 
int main()
{
    f();
    int t, n;
    scanf("%d", &t);
    while(t --){
        scanf("%d", &n);
        printf("%s\n", ok
?"no":"yes");
    }
}