清华大学考研复试机试：质因数的个数

题目描述

求正整数N(N>1)的质因数的个数。 相同的质因数需要重复计算。如120=2x2x2x3x5，共有5个质因数。

输入描述

可能有多组测试数据，每组测试数据的输入是一个正整数N，(1<N<10^9)。

输出描述

对于每组数据，输出N的质因数的个数。

分析

这道题最容易想到的一个思路就是，先求出小于N的所有质数，然后再遍历求出的质数，判断是否为N的因数。因为这道题的数据很大，就算求素数使用的是线性筛选素数的方法(不知道的可以去Google)，还是会超时。

下面给出线性筛选素数的模板代码：

#include<cstdio>
#include<cstring>
#define MAXN 100005
#define MAXL 1299710
int prime[MAXN];
int check[MAXL];

int main(void)
{
	int tot = 0;
	memset(check, 0, sizeof(check));
	for (int i = 2; i < MAXL; ++i)
	{
	  if (!check[i])
	  {
	    prime[tot++] = i;
	  }
	  for (int j = 0; j < tot; ++j)
	  {
	    if (i * prime[j] > MAXL)
	    {
	      break;
	    }
	    check[i*prime[j]] = 1;
	    if (i % prime[j] == 0)
	    {
	      break;
	    }
	  }
	}
	return 0;
}

线性筛素数的精髓在于，筛选过程中保证了两点：一是合数一定被删掉了；二是每个数都没有被重复的删掉，即每个数都只被它最小的质因数给删掉。这是比埃拉托斯特尼筛选法(简称埃法)更快的原因(这个方法应该听过吧)。证明的话请自行Google，如果证明看得不是很明白的话，可以自己手动筛选一下，应该就明白了。

上述方法之所以不行，是因为我们把求质数和求因数的过程分开计算，这道题需要我们同时计算质数和因数，这样就不会超时，那么怎么将这两个过程合并到一起呢？其实很简单，我们在求因数的时候，同时将N缩小。比如N的一个因数为2，我们就让N一直除以2，直到N不能被2整除，这样N就一定不会有2的倍数的因数存在，比如4，6这些合数，在这个过程中是不是也就变相地在筛选质数了呢！这样N的因数中一定不会有合数，我们求出的因数就一定是质因数，同时也算出了题目中所要求出的质因数的个数。如果是求质因数种类的个数，也可以使用这个方法。

理论上这个方法可行，复杂度也不高，不过实现上会有一些bug存在，所以AC代码写得有点奇怪(-_-||)。

AC代码如下：

#include<iostream>
using namespace std;

int main()
{
    int n;//输入
    int result;//输出
    while(cin >> n)
    {
        result = 0;
        //这种循环也实现了质因数，注意这里的循环条件是i<=n
        //如果不写循环里面的if语句的话，还是会超时
        //如果将循环条件改为i*i<=n，当N只有一种质因数时
        //即N为4,8,9,25这种数时，结果会出错
        //不过还是能够AC，因为好像清华给的数据里面没有这种数
        for(int i = 2; i <= n; ++i)
        {
            while(n % i == 0)
            {
                n /= i;
                ++result;
            }
            //举个例子，比如n为6，那么这样只会进入循环一次，就可以求出答案
            //如果数据很大，这样减少循环的次数很可观
            //如果有兴趣可以试一试删掉if判断，还是会超时的
            //n!=1的判断是用来解决上述N只有一种质因数时，答案错误的情况
            if(n < i*i && n != 1)
            {
                ++result;
                break;
            }
        }
        cout << result << endl;
    }
    return 0;
}