L1-028 判断素数

本题的目标很简单，就是判断一个给定的正整数是否素数。
输入格式：
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2³¹的需要判断的正整数。
输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。
输入样例：

2
11
111
结尾无空行

输出样例：

Yes
No
结尾无空行

AC代码：

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int num;
    cin >> num;
    while(num--)
    {
        int n;
        cin >> n;
        if(n == 1)
        {
            cout << "No";
        }
        else if(n == 2)
        {
            cout << "Yes";
        }
        else{
            int i;
            for(i = 2; i <= sqrt(n); i++)
            {
                if(n % i == 0)
                {
                    cout << "No";
                    break;
                }
            }
            if(i > sqrt(n))
            {
                cout << "Yes";
            }
        }
        if(num > 0)
        {
            cout << endl;
        }
    }
    
    return 0;
}