PTA测试有一条运行超时 不懂怎么修改代码🥲【要求是英文的 在下方图片中】

Question

#include <stdio.h>

int gx(int);
int fx(int);
int textprime(int);

int main()
{
int i, n, num, temp;
scanf("%d", &n);

for (i = 0; i < n; i++)
{
    scanf("%d", &num);
    temp = gx(num);
    
    if (temp == 1)
        printf("YES\n");
    else
        printf("NO\n");
}
return 0;

}

int textprime(int num)
{
int i, count = 0;
for (i = 1; i <= num; i++)
{
if (num % i == 0)
count++;
}
if (count == 2)
return 1;
else
return 0;
}

int fx(int num)
{
int i = 0;
while (num) // 8 9 10 11
{
i = textprime(num + 1);
if (i == 1)
{
return num + 1;
}
else
num++;
}
}

int gx(int num)
{
int state, result = 0;
result = (fx(num) + fx(fx(num))) / 2;

state = textprime(result);
return state;

}

Answer 1

判断素数的函数修改一下试试：

#include <math.h>


int textprime(int num)
{
    int i;
    int t = sqrt((double)num); //需要包含math.h
    if(num < 2) return 0;

    for (i = 2; i<=t; i++) 
    {
        if (num % i == 0)
            return 0;
    }
    return 1;
}