• Source
    1. /// Author : Shohahur Rahaman
    2. /// URI : 1558
    3.  
    4. #include<stdio.h>
    5. #define N 10000
    6.  
    7. int len=0;
    8. void pre_gen(int ara[]);
    9.  
    10. int main()
    11. {
    12. int n,i,ck;
    13. int ara[N+1] = {0};
    14. pre_gen(ara);
    15.  
    16. while(scanf("%d",&n)==1){
    17.  
    18. if(n >= 0 && ara[n]==1)
    19. printf("YES\n");
    20. else
    21. printf("NO\n");
    22. }
    23.  
    24. return 0;
    25. }
    26.  
    27. void pre_gen(int ara[])
    28. {
    29. int i,j,tmp;
    30.  
    31. for(i=0;i*i<=N;i++){
    32. for(j=0;j*j<=N;j++){
    33.  
    34. if( (i*i)+(j*j) <= N ){
    35. tmp = (i*i)+(j*j);
    36. ara[tmp] = 1;
    37. // printf("%d\n",ara[len]);
    38. }
    39.  
    40. }
    41. }
    42. }
    43.