【例题】【二项式定理】NKOJ3811 lucknum

NKOJ3811 lucknum
时间限制 : - MS 空间限制 : 165536 KB
评测说明 : 1000ms
问题描述
每个人都会有幸运数字，有种幸运数字是这样定义的：
如果X是幸运数字，则X在m进制下的表示为x1x2…xk，一定有x1<=x2<=…<=xk，其中k可以表示X在m进制下的位数（不能有前导0，除非该数本身就是0）。
这样的数字可能有无穷多个的，但是如果是在m进制下位数不超过n的幸运数字，就应该是有限个了，你能算出来吗？
这个答案可能很大，你只需要输出答案对一个质数p取模的值即可。

输入格式
共一行，三个正整数 n、m 和 p，保证 p 是质数。

输出格式
共一行，表示答案对p取模的值。

样例输入
4 10 10000079

样例输出
715

提示
前 20%的数据满足 n <= 18, m <= 10。
前 50%的数据满足 n <= 100, m <= 100。
前 80%的数据满足 n <= 1000, m <= 1000。
100%的数据满足 n <= 10^7, m <= 10^7, n + m <= p, p <= 10000079

观察发现该题即为求c[n+m-1][m-1]

#include<cstdio>
#include<iostream>
using namespace std;
const int need=1e7+5;

int n,m,p;
//.............................................................
int pr[need],tot,cnt[need],s[need];
bool mark[need];

void get_pr(int n)
{
    for(int i=2,j,temp;i<=n;i++)
    {
        if(!mark[i]) {pr[++tot]=i;s[i]=tot;}
        for(j=1;j<=tot&&(temp=i*pr[j])<=n;j++)
        {
            mark[temp]=true;
            s[temp]=j;
            if(i%pr[j]==0) break;
        }
    }
}

void add(int x,int val)
{
    while(x!=1)
    {
        cnt[s[x]]+=val;
        x/=pr[s[x]];
    }
}

int c_(int n,int m)
{
    get_pr(n);
    for(int i=m+1;i<=n;i++) add(i,1);
    for(int i=1,mn=n-m;i<=mn;i++) add(i,-1);
    int ans=1;
    for(int i=1;i<=tot;i++)
     while(cnt[i]--) ans=(long long)ans*pr[i]%p; 
    return ans;
}


int main()
{
    scanf("%d%d%d",&n,&m,&p);
    cout<<c_(n+m-1,m-1);
}

因为p为质数，所以：

#include<cstdio>
#include<iostream>
using namespace std;
const int need=1e7+5;

int n,m,p;
int d[need];

int power(int a,int b=p-2,int c=p)
{
    int ans=1;
    a%=c;
    while(b)
    {
        if(b&1) ans=(long long)ans*a%c;
        b>>=1;
        a=(long long)a*a%c;
    }
    return ans;
}


int main()
{
    scanf("%d%d%d",&n,&m,&p);
    d[1]=1;
    for(int i=2,nn=m+n-1;i<=nn;i++) d[i]=((long long)d[i-1]*i)%p;
    int inv_n=power(d[n]);
    int inv_mn=power(d[m-1]);
    printf("%d",((long long)d[n+m-1]*inv_n)%p*inv_mn%p);
}