hdu 4390 Number Sequence（容斥原理）

题目：给出n个数，b1,b2,b3……bn，构造n个数，求有多少组序列a1,a2,……an(ai>1)，使得a1*a2*a3……an=b1*b2……bn（bi>1）;
http://acm.hdu.edu.cn/showproblem.php?pid=4390
首先是将所有 的b进行素因子分解，则a和b的因子是完全一致的。
剩下的便是将所有b的因子，分给a
我们考虑某个素因子pi,如果有ci个，便成了子问题将ci个相同的物品放入到n个不同的容器中，种数为多少
但是要求ai>1，也就是容器不能为空，这是个问题。
我们考虑的是什么的情况，然后减去假设有一个确定是空的情况，发现可以用容斥原理解决
我们假设f[i]表示有i个容器的结果，c(n,i)*f[i]

将m个物品放到到不同的n个容器中，结果为c(n+m-1,n-1)

#include<iostream>
#include<cstdio>
#include<math.h>
#include<vector>
#include<map>
#include<string.h>
using namespace std;

#define mod 1000000007
#define N 1000001
typedef __int64 LL;

int b,a[N],prime[N/3],num[N/3];
LL c[600][600];
void inint()
{
    int i,j,w=0,k;
    memset(a,0,sizeof(a));
    for(i=2; i<N; i++)
    {
        if(!a[i]) prime[w++]=i;
        for(j=0; j<w&&(k=i*prime[j])<N; j++)
        {
            a[k]=1;
            if(i%prime[j]==0) break;
        }
    }

    for(i=0; i<600; i++)
    {
        c[i][0]=c[i][i]=1;
        for(j=1; j<i; j++)
        c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
    }
}

LL gets(int n,int m)
{
    return c[n+m-1][n-1];
}
map<int,int> p;
map<int,int>::iterator iter;
int main()
{
    int i,j,n,bit,k;
    inint();
    LL res,cent;
    while(~scanf("%d",&n))
    {
        p.clear();
        for(i=0; i<n; i++)
        {
            scanf("%d",&b);
            for(j=0; prime[j]<=b; j++)
            {
                while(b%prime[j]==0)
                {
                    if(p.count(prime[j])<0) p[prime[j]]=1;
                    else p[prime[j]]++;
                    b=b/prime[j];
                }
            }
        }
        cent=p.size();
        res=1;
        k=0;
        num[0]=1;
        for(iter=p.begin();iter!=p.end();iter++)
        {
            num[k]=iter->second;
            res=(res*gets(n,num[k]))%mod;
            k++;
        }
        for(i=1;i<n;i++)
        {
            cent=c[n][i];
            for(j=0;j<k;j++)
            {
                cent=(cent*gets(n-i,num[j]))%mod;
            }
            if(i&1)
            res=((res-cent)%mod+mod)%mod;
            else
            res=((res+cent)%mod+mod)%mod;
        }
        printf("%lld\n",res);
    }
    return 0;
}