洛谷P3746 [六省联考 2017] 组合数问题

题目描述

组合数 𝐶𝑛𝑚Cnm​ 表示的是从 𝑛n 个互不相同的物品中选出 𝑚m 个物品的方案数。举个例子， 从 (1,2,3)(1,2,3) 三个物品中选择两个物品可以有 (1,2)(1,2)，(1,3)(1,3)，(2,3)(2,3) 这三种选择方法。根据组合数的定义，我们可以给出计算组合数 𝐶𝑛𝑚Cnm​ 的一般公式：

𝐶𝑛𝑚=𝑛!𝑚! (𝑛−𝑚)!Cnm​=m! (n−m)!n!​

其中 𝑛!=1×2×⋯×𝑛n!=1×2×⋯×n。（特别地，当 𝑛=0n=0 时，𝑛!=1n!=1；当 𝑚>𝑛m>n 时，𝐶𝑛𝑚=0Cnm​=0。）

小葱在 NOIP 的时候学习了 𝐶𝑖𝑗Cij​ 和 𝑘k 的倍数关系，现在他想更进一步，研究更多关于组合数的性质。小葱发现，𝐶𝑖𝑗Cij​ 是否是 𝑘k 的倍数，取决于 𝐶𝑖𝑗 mod 𝑘Cij​modk 是否等于 00，这个神奇的性质引发了小葱对 modmod 运算（取余数运算）的兴趣。现在小葱选择了是四个整数 𝑛,𝑝,𝑘,𝑟n,p,k,r，他希望知道

(∑𝑖=0∞𝐶𝑛𝑘𝑖𝑘+𝑟) mod 𝑝,(i=0∑∞​Cnkik+r​)modp,

即

(𝐶𝑛𝑘𝑟+𝐶𝑛𝑘𝑘+𝑟+𝐶𝑛𝑘2𝑘+𝑟+⋯+𝐶𝑛𝑘(𝑛−1)𝑘+𝑟+𝐶𝑛𝑘𝑛𝑘+𝑟+⋯ ) mod 𝑝(Cnkr​+Cnkk+r​+Cnk2k+r​+⋯+Cnk(n−1)k+r​+Cnknk+r​+⋯)modp

的值。

输入格式

第一行有四个整数 𝑛,𝑝,𝑘,𝑟n,p,k,r，所有整数含义见问题描述。

输出格式

一行一个整数代表答案。

输入输出样例

输入 #1

2 10007 2 0

输出 #1

8

输入 #2

20 10007 20 0

输出 #2

176

说明/提示

对于 30%30% 的测试点，1≤𝑛,𝑘≤301≤n,k≤30，𝑝p 是质数；
对于另外 5%5% 的测试点，𝑝=2p=2；
对于另外 5%5% 的测试点，𝑘=1k=1；
对于另外 10%10% 的测试点，𝑘=2k=2；
对于另外 15%15% 的测试点，1≤𝑛≤103,1≤𝑘≤501≤n≤103,1≤k≤50，𝑝p 是质数；
对于另外 15%15% 的测试点，1≤𝑛×𝑘≤1061≤n×k≤106，𝑝p 是质数；
对于另外 10%10% 的测试点，1≤𝑛≤109,1≤𝑘≤501≤n≤109,1≤k≤50，𝑝p 是质数；
对于 100%100% 的测试点，1≤𝑛≤109,0≤𝑟<𝑘≤50,2≤𝑝≤230−11≤n≤109,0≤r<k≤50,2≤p≤230−1。

Code:

#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
#define int long long 
using namespace std;
const int MAXN=1e6;
inline int read()
{
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int N,mod,k,r;
int C[51][51];
struct Matrix
{
    int a[51][51];
    Matrix(){memset(a,0,sizeof(a));}
};
Matrix mul(Matrix x,Matrix y)
{
    Matrix c;
    for(int kk=0;kk<=k-1;kk++)
        for(int i=0;i<=k-1;i++)
            for(int j=0;j<=k-1;j++)
                c.a[i][j]=(c.a[i][j]+x.a[i][kk]*y.a[kk][j]%mod)%mod;
    return c;
}
void out(Matrix x)
{
    for(int i=0;i<=k-1;i++,puts("\n"))
        for(int j=0;j<=k-1;j++)
            printf("%d ",x.a[i][j]);
}
Matrix fastpow(Matrix a,int p)
{
    Matrix base;
    for(int i=0;i<=k;i++) base.a[i][i]=1;
    while(p)
    {
        if(p&1) base=mul(base,a);
        a=mul(a,a);
        p>>=1;
    }
    return base;
}
main()
{
    #ifdef WIN32
    freopen("a.in","r",stdin);
    #endif
    N=read(),mod=read(),k=read(),r=read();
    Matrix tmp;
    for(int i=0;i<=k-2;i++)
        tmp.a[i][i]=tmp.a[i][i+1]=1;
    tmp.a[k-1][0]++;tmp.a[k-1][k-1]++;
    Matrix ans;
    ans.a[0][0]=1;
    tmp=fastpow(tmp,N*k);
    ans=mul(ans,tmp);
    printf("%lld",ans.a[0][r]);
    return 0;
}