题解 | 判断质数

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

bool check(int a){
	if(a == 1) return 0;
	else if(a == 2 || a == 3) return 1;
	else if(a % 2 == 0) return 0;
	else{
		for(int i=2 ; i <= sqrt(a) ; i++){
			if(a % i == 0){
				return 0;
				break;
			}
		}
		return 1;
	}
}

int main(){
	int n;  cin >> n;
	
	if(check(n)) cout << "Yes";
	else		cout << "No";
	
	return 0;
}