POJ 2603 Brave balloonists（素因子分解）

Description
求十个数乘积的因子个数
Input
十个整数（1<=ai<=10000)
Output
输出十个整数乘积的因子个数
Sample Input
1
2
6
1
3
1
1
1
1
1
Sample Output
9
Solution
简单数论题，对每个数素因子分解得到每个素因子pi的幂次ai，那么因子个数即为(1+a1)(1+a2)…*(1+am)
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
#define maxn 11111
int prime[maxn],is_prime[maxn],res;
map<int,int>factor;
void get_prime()
{
    memset(is_prime,0,sizeof(is_prime));
    res=0;
    for(int i=2;i<maxn;i++)
        if(!is_prime[i])
        {
            prime[res++]=i;
            for(int j=i;j<maxn;j+=i)
                is_prime[i]=1;
        }
}
void get_factor(int n)
{
    for(int i=0;i<res&&prime[i]<=n;i++)
        if(n%prime[i]==0)
            while(n%prime[i]==0)
                factor[prime[i]]++,n/=prime[i];
    if(n!=1)factor[n]++;
}
int main()
{
    get_prime();
    factor.clear();
    for(int i=0;i<10;i++)
    {
        int n;
        scanf("%d",&n);
        get_factor(n);
    }
    int ans=1;
    for(map<int,int>::iterator it=factor.begin();it!=factor.end();it++)
        ans=(ans*(it->second+1))%10;
    printf("%d\n",ans);
    return 0;
}