求素数

#include<iostream>
#include<string.h>
#include<math.h>
using namespace std;

int main(){
    int n,flag;
    while (cin >> n)
    {
        flag = 0;
        if (n <= 1)
            flag = 1;
        int max = sqrt(n) + 1;
        for (int i = 2; i <= max; i++)
            if (n%i == 0)
            {
                flag = 1; break;
            }
        puts((flag == 1) ? "no" : "yes");
    }
    
    return 0;
}