0

I've been doing a few of the challenges on the Sphere Online Judge (SPOJ), but I can't seem to get the second problem (the prime generator) to run within the time limit. How can the speed of the following code be increased?

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

int is_prime(int n);
void make_sieve();
void fast_prime(int n);

int primes[16000];

int main()
{
    int nlines;
    int m, n;
    make_sieve();
    scanf("%d", &nlines);
    for (; nlines >= 1; nlines--) {
        scanf("%d %d", &m, &n);
        if (!(m % 2)) {
            m++;
        }
        for ( ; m < n; m+=2) {
            fast_prime(m);
        }
        printf("\n");
    }
    return 0;
}

/* Prints a number if it's prime. */
inline void fast_prime(int n)
{
    int j;
    for (int i = 0; ((j = primes[i]) > -1); i++) {
        if (!(n % j)) {
            return;
        }
    }
    printf("%d\n", n);
}

/* Create an array listing prime numbers. */
void make_sieve()
{
    int j = 0;
    for (int i = 0; i < 16000; i++) {
        primes[i] = -1;
    }
    for (int i = 2; i < 32000; i++) {
        if (i % 2) {
            if (is_prime(i)) {
                primes[j] = i;
                j++;
            }
        }
    }
    return;
}

/* Test if a number is prime. Return 1 if prime. Return 0 if not. */
int is_prime(int n)
{
    int rootofn;
    rootofn = sqrt(n);
    if ((n <= 2) || (n == 3) || (n == 5) || (n == 7)) {
        return 1;
    }
    if (((n % 2) == 0) || ((n % 3) == 0) || ((n % 5) == 0) || ((n % 7) == 0)) {
        return 0;
    }
    for (int i = 11; i < rootofn; i += 2) {
        if ((n % i) == 0) {
            return 0;
        }
    }
    return 1;
}
4

2 回答 2

2

isprime() 不使用素数表 primes[]。

另外,实现素数数组的搜索,该搜索将使用二分搜索算法快速完成。标准库有一个。

要查看您在代码中花费的时间,您可以使用 profiling gcc 示例

gcc -p -g - o mycode mycode.c
===run the code--
gprof mycode
于 2010-05-29T09:57:04.580 回答
1

目前,您的问题不是时间限制。这是您的程序从不打印任何数字的事实。

最明显的错误是,在 fast_prime 中,您正在检查 n 是否可以被 prime[0]、prime[1]、... 直到 prime[k] 整除。即使n是素数,你也不会打印它,因为n在素数[]中的某个地方,所以你会得到n可以被某个数整除......

要纠正这个问题,您需要检查 n 是否可以被某个素数整除,直到 n 的平方根(这也会产生加速代码的副作用,因为在确定某个数字是素数之前将检查较少的数字)

将 fast_prime 更改为

inline void fast_prime(int n)
{    
  int j;
  int rootofn;
  rootofn = sqrt(n);        
  for (int i = 0; ((j = primes[i]) > -1) && (j<rootofn); i++) {
      if (!(n % j)) {
          return;
      }
  }
  printf("%d\n", n);
}
于 2010-05-29T09:33:31.830 回答