[数论+素数判定] 试除法判定质数(基础+边界问题+代码风格)

1. 试除法+代码风格

866. 试除法判定质数

质数定义： 质数是指在大于 1 的自然数中，除了 1 和它本身以外不再有其他因数的自然数。

试除法判定质数思路：

• 最基础：从 2 ~ n - 1 依次枚举查看能否整除，即存在了其它因子，则不为素数。时间复杂度 $O(n)$。
• 算法优化：因子总是成对出现，则只需要枚举到 $\sqrt{n}$ 即可。时间复杂度为 $O(\sqrt{n})$。
• 代码优化：不要使用 $i<=sqrt(n)$，因为 $sqrt()$ 函数很慢。也不要使用 $i\ast i<=n$ 因为可能存在 $(i+1)\ast (i+1)$ 刚好越界，那么直接成负数进行死循环了。优秀的写法应该为：$i<=n/i$

基础代码：

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

代码优化：

#include <iostream>
#include <algorithm>

using namespace std;

int m;

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

int main() {
    cin >> m;
    while (m --) {
        int n;
        cin >> n;
    
        if (is_prime(n)) puts("Yes");
        else puts("No");
    }
    
    return 0;
}