NKOJ-Unknow 幸运数字

幸运数字
题目描述

每个人都会有幸运数字， 有种幸运数字是这样定义的：
如果 X 是幸运数字， 则 X 在 m 进制下的表示为 x1x2...xk， 一定有 x1<=x2<=...<=xk， 其中 k 可以表示 X 在 m 进制下的位数。
这样的数字可能有无穷多个的， 但是如果是在 m 进制下位数不超过 n 的幸运数字， 就
应该是有限个了， 你能算出来吗？
这个答案可能很大， 你只需要输出答案对一个数 p 取模的值即可。

输入格式

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

输出格式

共一行， 表示答案对 p 取模的值。

样例输入 1

4 3 23

样例输出 1

15

样例输入 2

4 10 10000079

样例输出 2

715

样例说明

样例 1 的 15 种方案如下：
0000,0001,0011,0111,1111,0002,0022,0222,2222,0012,0112,1222,1112,1122,1222

数据范围

前 20%的数据满足 n <= 18, m <= 10。
前 40%的数据满足 n <= 100, m <= 100。
前 60%的数据满足 n <= 1000, m <= 1000。
100%的数据满足 n <= 10^7, m <= 10^7, n + m <= p, p <= 10000079

那这个人岂不是有无数个幸运数字，买彩票的时候不得纠结死

题解

分析

对于n=1

m=0 res=1
m=1 res=2
m=2 res=3

对于n=2

m=0 res=1
m=1 res=4
m=2 res=10

对于p=3

m=0 res=1
m=1 res=6
m=2 res=21

根据各种奇葩的打表得

对n，m，p，res=C[n+m-1][m-1]%mod

具体的过程我就不展示了…反正大家做不出来就根据结论去推过程

然后就是用卢卡斯定理求C[n+m-1][m-1]%mod

我知道这种直接摆结论实在是很弱，但是我也确实是很难求解…

附上对拍代码

#include <iostream>
#include <cstdio>
using namespace std;

long long mod;

long long KSM(long long a,long long b)
{
    long long res=1;
    while(b)
    {
        if(b&1)res=(res*a)%mod;
        b>>=1;
        a=(a*a)%mod;
    }
    return res;
}

long long C(long long a,long long b)
{
    if(a<b)return 0;
    if(a==b)return 1;
    if(b>a-b)b=a-b;
    long long A=1,B=1;
    for(int i=0;i<b;i++)
    {
        A=(A*(a-i))%mod;
        B=(B*(b-i))%mod;
    }
    return A*KSM(B,mod-2)%mod;
}

long long Lucas(long long n,long long m)
{
    if(m==0)return 1;
    return C(n%mod,m%mod)*Lucas(n/mod,m/mod)%mod;
}

int main()
{
    freopen("lucknum.in","r",stdin);
    freopen("lucknum.out","w",stdout);
    long long n,m;
    scanf("%lld%lld%lld",&n,&m,&mod);
    printf("%lld",Lucas(n+m-1,m-1));
}