首页 > 代码库 > uva 1415 - Gauss Prime(高斯素数)

uva 1415 - Gauss Prime(高斯素数)

题目链接:uva 1415 - Gauss Prime

题目大意:给出一个a,b,表示高斯数a+bi(i=?2,判断该数是否为高斯素数。

解题思路;

  • a = 0 时,肯定不是高斯素数
  • a != 0时,判断a2+2b2是否为素数即可。
#include <cstdio>
#include <cstring>
#include <cmath>

bool is_prime (int n) {
    int m = sqrt(n+0.5);
    for (int i = 2; i <= m; i++)
        if (n % i == 0)
            return false;
    return true;
}

bool judge (int a, int b) {
    if (a == 0)
        return false;
    return is_prime(a*a+2*b*b);
}

int main () {
    int cas;
    scanf("%d", &cas);
    while (cas--) {
        int a, b;
        scanf("%d%d", &a, &b);
        printf("%s\n", judge(a, b) ? "Yes" : "No");
    }
    return 0;
}