试题:求1000000以内的素数

之前无聊时, 写的找素数的代码, 记得当年考二级C时, 有这么一道上机题, 那时候求稳, 效率肯定不高, 所以今天用了排除法来做

先给出总的测试代码:

#include <cmath>
#include <ctime>
#include <iostream>
using namespace std;

#define MAX_DATA 1000000

int size;
int data[MAX_DATA / 2 + 1];

void prime1()
{
    ......
}

void prime2()
{
    ......
}

void prime3()
{
    ......
}

void output()
{
    for (int i = 0; i < size; i++) {
        cout << data[i] << " ";
        if (9 == i % 10) {
            cout << endl;
        }
    }
}

int main()
{
    prime1();
    prime2();
    prime3();
    cin.get();
    output();
    return 0;
}

下面是写的第一份实现代码:

void prime1()
{
    time_t start = time(NULL);

    size = 0;
    for (int i = 2; i <= MAX_DATA; i++) {
        int j = 0;
        while (j < size) {
            if (0 == i % data[j]) {
                break;
            }
            else {
                ++j;
            }
        }
        if (j == size) {
            data[size] = i;
            ++size;
        }
    }

    time_t finish = time(NULL);
    cout << "way 1 get " << size << " prime use " << (finish - start) << " seconds" << endl;
}

prime1()的运行时间约31s

这时我想是不是太没效率了
为了证实我的预感, 我用一般的思路写了第二份代码来比较, 这份代码没有用到排除法:

void prime2()
{
    time_t start = time(NULL);

    size = 0;
    for (int i = 2; i <= MAX_DATA; i++) {
        int j = 2;
        int k = (int)sqrt(i) + 1;
        while (j < k) {
            if (i % j == 0) {
                break;
            }
            else {
                ++j;
            }
        }
        if (j == k) {
            data[size] = i;
            ++size;
        }
    }

    time_t finish = time(NULL);
    cout << "way 2 get " << size << " prime use " << (finish - start) << " seconds" << endl;
}

prime2()的运行时间约为3s

比较效率,prime1()的排除法竟然比prime2()慢了一个数量级, 是什么原因影响了结果呢
很明显, sqrt()使得内嵌的循环要少做至少一半的检查, 这就是效率差距的根本所在

但我很不喜欢prime2()的实现, 原因有两点:
1. sqrt()是库函数, 是别人的东西, 我不太喜欢用别人现成的东西, 除非没办法
2. prime2()因为没有用排除法而多做了一些非必要的检查, 比如23这个数, 用prime2()时, 我们要让它对2, 3, 4, 5做检查, 但是我们知道, 如果23不能整除2, 那么它必然也不能整除4, 所以prime2()还是做了多余的事

若想兼得prime1()与prime2()的优势, 那么必须在prime1()中也用到sqrt()的思想, 但我说了, 我不太喜欢用别人的东西, 所以, 我在想怎么自己去实现一下sqrt()这个函数, 大学的<<数值计算>>中一定有这个东西的, 也许<<数学分析>>上也有, 但我实在是一点都不记得如何做了, 不过应该可以从网上找到一些相关的什么公式, 但想想还是做罢, 太难了, 我也没这心情去做这种事, 再者, 我也不可能实现得比编译器自带的这个库函数有效率, 那么不取平方根, 反之, 取平方应该就可以了

于是有了第三份代码：

void prime3()
{
    time_t start = time(NULL);

    size = 0;
    for (int i = 2; i <= MAX_DATA; i++) {
        int j = 0;
        while (j < size) {
            if (0 == i % data[j]) {
                break;
            }
            else if (i < (data[j] * data[j])) {
                j = size;
                break;
            }
            else {
                ++j;
            }
        }
        if (j == size) {
            data[size] = i;
            ++size;
        }
    }

    time_t finish = time(NULL);
    cout << "way 3 get " << size << " prime use " << (finish - start) << " seconds" << endl;
}

prime3()的运行时间约为1s, OK

当然, 兴许还有可优化的地方, 比如可以用一个数组来保存素数数组元素的平方(大数的平方会溢出), 但一定是要放到需要用到这个平方的时候才去计算和保存这个平方数(延迟计算), 这样的效率才可能高起来, 经测试只会快一点点, 也许还有更好的方案

前面说的排除法, 具体的术语是什么不记得了, 反正依赖于这样一个原理:
任意合数都可分解为素数之间的乘积(如12 = 2 * 2 * 3, 51 = 3 * 17)