PTA 判断素数 C语言非c++

本题的目标很简单，就是判断一个给定的正整数是否素数。

CSDN pta 判断素数 C语言非c++
素数判断的正确打开方式
输入格式：输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2​31​​ 的需要判断的正整数。输出格式：对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。
输入样例：
2
11
111
输出样例：
Yes
No

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

int main() {
	int N, inputNum, flag,i;
	scanf("%u", & N);
	while (N--)
	{
		flag = 0;
		scanf("%u", &inputNum);
		if (inputNum > 1)
			flag = 1;
		for (i = 2; i <=sqrt(inputNum);i++)
			if (inputNum % i == 0)
			{
				flag = 0;
				break;
			}
		if (flag || inputNum == 2)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}