（PAT）统计给定区间内的三位数中有两位数字相同的完全平方数（如144、676）的个数

本题要求实现一个函数，统计给定区间内的三位数中有两位数字相同的完全平方数（如144、676）的个数。

函数接口定义：
int search( int n );

其中传入的参数int n是一个三位数的正整数（最高位数字非0）。函数search返回[101, n]区间内所有满足条件的数的个数。

裁判测试程序样例：
#include <stdio.h>
#include <math.h>

int search( int n );

int main()
{
int number;

scanf("%d",&number);
printf("count=%d\n",search(number));
	
return 0;

}

/* 你的代码将被嵌在这里 */

输入样例：
500

输出样例：
count=6

#include <stdio.h>
#include <math.h>

int search( int n );

int main()
{
    int number;

    scanf("%d",&number);
    printf("count=%d\n",search(number));
		
    return 0;
}
int search(int n)
{
	int i,cnt=0;
	for(i=101;i<=n;i++){
		double digit=sqrt(i);
		int a=i/100;
		int b=i/10%10;
		int c=i%10;
		if((digit==(int)digit)&&(a==b||a==c||b==c)) 
		cnt++;
	}
	return cnt;
}