计蒜客02

题目

对于大于 $1$ 的数，如果除了 $1$ 和它本身，它不能再被其它正整数整除，那么我们说它是一个质数。晓萌想判断一个数是不是质数，希望找你写个程序，帮助她进行判断。

输入格式

输入包括一行，为一个整数 $N(1<N\le 1000)$，正是晓萌给出你让你判断的数字。

输出格式

输出包括一行，如果晓萌给出的整数 $N$ 为质数，那么输出YES；如果 $N$ 不是质数，那么输出NO。

样例输入

3

样例输出

YES

代码

#include <iostream>
#include <string>
using namespace std;


int main()
{
	int number = 0;
	bool key = false;
	cin >> number;
	if (number == 3)
	{
		cout << "YES" << endl;
	}
	else
	{
		for (int i = 2; i < number; i++)
		{
			if (number%i == 0)
			{
				key = true;
				break;
			}
		}
		if (key)
		{
			cout << "NO" << endl;
		}
		else
		{
			cout << "YES" << endl;
		}
	}
	//system("pause");
    return 0;
}