PTA - L1-028 判断素数

#include <iostream>
#include <cmath>
using namespace std;

unsigned int InNum = 0;
void Is_sNum(int);

int main()
{
	cin >> InNum; int x = 0;

	while (InNum--)
	{
		cin >> x;
		Is_sNum(x);
	}
	return 0;
}

void Is_sNum(int num)
{
    if(1 == num)
    {
        cout << "No" << endl;
        return;
    }
    
	if (2 == num)
	{
		cout << "Yes" << endl;
        return;
	}

	int y = sqrt(num);
	for (size_t i = 2; i <= y ; i++) 
	{
		if (num % i == 0) 
		{
			cout << "No" << endl;
			return;
		}
	}
	cout << "Yes" << endl;
	return;
}