Dev Skill DCP-27: Is It Prime? Solution (Check Prime or Not Prime)

Problem Link
Dev Skill DCP-27: Is It Prime? Solution

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