如何判断质数

描述

输入n个整数m，判断它们是否为质数。如果是输出“YES”，否则输出“NO”（1≤n≤50）。

输入

输入文件有n+1行数据，第一行为n，

第2-n+1行每行为一个正整数m（0≤m≤10000000）。

输出

输出文件有n行数据，每行分别是“YES”或者“NO”。

输入样例 1 

3
8
9
13

输出样例 1

NO
NO
YES

#include<bits/stdc++.h>
using namespace std;
int a(int n) {
	if(n==1) {
		return 0;
	}
	for(int i=2; i<n; i++) {
		if(n%i==0)
			return 0;
	}
	if(n==1) {
		return 0;
	}
	return 1;
}
int main() {
	int m,s;
	cin>>s;
	for(int i=1; i<=s; i++) {
		cin>>m;
		if(a(m)==1) {
			cout<<"YES"<<endl;
		} else {
			cout<<"NO"<<endl;
		}
	}
	return 0;
}