计算机历年考研复试上机题------质因数的个数

题目传送门

题目描述
求正整数N(N>1)的质因数的个数。 相同的质因数需要重复计算。如>120=2*2*2*3*5，共有5个质因数。
输入描述:
可能有多组测试数据，每组测试数据的输入是一个正整数N，(1< N< 10^9)。
输出描述:
对于每组数据，输出N的质因数的个数。
示例1
输入
120
输出
5

这道题和前面的一道计算机历年考研复试上机题——约数个数其实是一个类型的题目，只是这里只要求求出质因数的个数。我们只需要利用 约数定理 求出我们需要的质因数的幂就行了。同时注意大于根号n可能会有一个约数就行了。

下面是代码：

#include<iostream>
using namespace std;
int main()
{
    int num;
    while (~scanf("%d", &num))
    {
        int ans = 0;
        for (int i = 2; i*i < num; i++)
        {
            while (num%i == 0)
            {
                ans++;
                num = num / i;
            }
        }
        if (num > 1) ans++;
        printf("%d\n",ans);
    }
    return 0;
}

以下是我想的测试用例：

输入：
5
输出：
1