使用函数判断完全平方数

#include<stdio.h>
#include<math.h>
int IsSquare(int n)
{
  int i,a=0;
  for(i=1;i*i<=n;i++)
  {
    if(i*i==n)
    {
      a=1;
      break;
    }
    
  }
  return a;
}
int main()
{
  int n,a;
  scanf("%d",&n);
  a=IsSquare(n);
  if(a)
  {
    printf("YES");
  }
  else 
  printf("NO");
  return 0;
}