7-4 判断素数 (10分)

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int N;
    cin >> N;
    while (N--)
    {
        int p,flag=0;
        cin >> p;
        if (p == 1)flag = 1;
        for (int i = 2; i <= sqrt(p); i++)
        {
            if (p % i == 0)
            {
                flag = 1;
                break;
            }
        }
        !flag?cout << "Yes" << endl:cout << "No" << endl;
    }
    return 0;
}