Проверка чисел на простоту

#include "stdafx.h"
#include "iostream"
#include "vector"
 
using namespace std;
 
int f(int n) {
        vector<char> prime (n+1, true);
prime[0] = prime[1] = false;
for (int i=2; i<=n; ++i)
    if (prime[i])
        for (int j=i*i; j<=n; j+=i)
            prime[j] = false;
return prime[n];}
 
int main () {
    int n;
    cin >> n;
    int *a=new int [n];
    for (int i=0;i<n;i++) {cin >> a[i];
    if (f(a[i])) cout << "YES" << endl; else cout << "NO" << endl;}
 
    
return 0;}

#include <iostream> 
#include <cmath> 
 
int prost(int x)
 {   if (x==2) return 1;
     if (x==0||x==1||x%2==0) return 0;
     for (int i=3; i<sqrt((float)x); i+=2)
        if (x/i==0) return 0;
        return 1;
 }
 
void main ()
{ int *A,N;
std::cout<<"Input N ";
std::cin>>N;
A=new int [N];
for (int i=0; i<N; i++)
    {   std::cout<<"A["<<i<<"]"<<std::endl;
        std::cin>>A[i];
        if (prost(A[i])) std::cout<<"yes"<<std::endl;
        else std::cout<<"no"<<std::endl;
    }
system ("pause");
}

#include <iostream>
#include <cmath>
 
bool isPrime(unsigned int number)
{
   if (number < 2)
      return 0;
   unsigned int root = static_cast<unsigned int>(std::sqrt(static_cast<double>(number)));
   for (int i = 2; i <= root; i++)
      if ((number % i) == 0)
         return 0;
   return 1;
}
 
int main(int argc, char* argv[])
{
   unsigned int size, num;
 
   std::cout << "Ââåäèòå êîëè÷åñòâî ÷èñåë: ";
   std::cin >> size;
 
   for (int i = 0; i < size; i++)
   {
 
      std::cout << "Ââåäèòå ÷èñëî: ";
      std::cin >> num;
      if (isPrime(num))
         std::cout << "×èñëî " << num << " ïðîñòîå." << std::endl;
      else
         std::cout << "×èñëî " << num << " ñîñòàâíîå." << std::endl;
   }
   std::system("pause");
   return 0;
}

if (x==0||x==1||x%2==0) return 0;

if (x<2||x%2==0) return 0;