判断满足条件的三位数

本题要求实现一个函数，统计给定区间内的三位数中有两位数字相同的完全平方数（如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

int search(int n) {
    int count = 0;
    for (int i = 101; i <= n; i++) {
        int sqrt_i = sqrt(i);
        if (sqrt_i * sqrt_i == i) { // i 是完全平方数
            int last_digit = i % 10; // 取出个位数字
            int second_last_digit = (i / 10) % 10; // 取出十位数字
            int third_last_digit = i / 100; // 取出百位数字
            if (last_digit == second_last_digit || 
                last_digit == third_last_digit || 
                second_last_digit == third_last_digit) {
                count++;
            }
        }
    }
    return count;
}

供参考：

#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 cnt = 0, i, t;
    if (n <= 100 || n > 999)
        return cnt;
    for (i = 101;i <= n; i++){
         t = (int)sqrt(i);
         t *= t;
         if (t == i && (i % 10 == i / 100 || i % 10 == i / 10 % 10 || i / 100 == i / 10 % 10))
             cnt++;
    }
    return cnt;
}