kAri 399 复数素数判断

397. Prime Judge

时间限制 1000 ms  内存限制 65536 KB

题目描述

众所周知，如果一个正整数只能被1和自身整除，那么该数被称为素数。题目的任务很简单，就是判定一个数是否是一个素数。 只不过可能数的形式与正整数有一些不同，数的形式为a+bi，其中a、b为整数，且ii被定义为-1。如果a+bi能被分解为(a1+b1i)(a2+b2i)的形式，那么该数不是素数；否则，该数是素数。其中a1 、b1、 a2 、b2均为整数，且1，-1，0，i，-i不能作为被分解的因子。 注意，1，-1，0，i，-i均不为素数。

输入格式

输入包括若干组数据，每行包含2个数a、b，表示一个形如a+bi的数。a，b小于10000。

输出格式

对应于输入每行，如果输入的数为素数，则输出“YES”，否则输出“NO”。（不包括引号）

输入样例

-10 2
3 0

输出样例

NO
YES

代码：
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
/*判断a+bi是否为素数
复数，|A*B|=|A|*|B|
则，a*a+b*b=(a1*a1+b1*b1)(a2*a2+b2*b2)
依据此条件暴力搜索
*/
int is_sqt(int a)
{
    if(fabs(floor(sqrt(a))*floor(sqrt(a))-a)<0.0000001)
        return 1;
    else
        return 0;
}
int main()
{
    int n,a,b,i,j,k,yes;
    while(scanf("%d %d",&a,&b)!=EOF)
    {
        n=a*a+b*b;
        yes=0;
        //题中特别说明，-1,1,i,-i不是素数
        if(n==1||n==0)
            printf("NO\n");
        else
        {
            //枚举n的约数
            for(i=2;i<sqrt(n)+0.5&&yes==0;i++)
            {
                if(n%i==0)
                {
                    //枚举a1  用平方判边界，每个数开方太繁
                    for(j=0;j*j<=i/2&&yes==0;j++) //注意初始条件
                    {
                        //若满足为平方和，则枚举a2
                        if(is_sqt(i-j*j))
                        {
                            for(k=0;k*k<=n/i/2&&yes==0;k++)
                            {
                                //若满足平方和，则为非素数
                                if(is_sqt(n/i-k*k))
                                {
                                    yes=1;
                                }
                            }
                        }
                    }
                }
            }
            if(yes==1)
                printf("NO\n");
            else
                printf("YES\n");
        }
    }
    return 0;
}