輸入一個數n,判斷n是不是素數。

#include<iostream>
using namespace std;
int main()
{
	int  n;          //判斷n是不是素數,n>0。 
	cin>>n;
	cout<<endl;
	int i=2;          
	int k;
	if(n==1)
	k=0;
	else if(n==2)
	k=1;
	else
	{
		while(i<n)     //可以用for(i=2;i<n;i++)
			if(n%i==0)
			{
			k=0;
			break;	
			}
		if(i==n)
		k=1;
	
		
	}
	
	if(k==1)
	cout<<n<<" is Prime number"<<endl;
	  else
	  cout<<n<<" is not Prime number"<<endl;
	return 0;
}