例题(4.2) 简单素数判断

题目-例题(4.2) 简单素数判断

来源

北京大学 计算概论 06

描述

对于自然数n（0 < n < 10），判断n是否是素数。

关于输入

输入仅一行，一个自然数n（0 < n < 10）。

关于输出

输出仅一行，如果n是素数，输出yes；否则，输出no。

例子输入

7

例子输出

yes

提示

0到10之间的素数仅有四个：2，3，5，7。

#include <stdio.h>
#include <math.h>

int isPrime(int n)
{
	if(n < 6)
	{
		if(n == 1 || n == 4)
		{
			return 0;
		}
		else
		{
			return 1;
		}
	}
	else /* if(n >= 6) */
	{
		int r = n%6;
		if(r == 1 || r == 5) /* maybe prime */
		{
			int d, y;
			y = (int) sqrt((double) n);
			for(d = 6; d <= y; d += 6)
			{
				if(n%(d - 1) == 0 || n%(d + 1) == 0)
				{
					return 0;
				}
			}
			return 1;
		}
		else
		{
			return 0;
		}
	}
}

int main()
{
	int n;
	const char *answer[] = {"no", "yes"};
	scanf("%d", &n);
	printf("%s\n", answer[isPrime(n)]);

	return 0;
}