題目的大意就是要求我們從n*k個數字中選擇%k餘r的方案數

定義f(i,j)=∑∞t=0Cj+tki
f(i,j)=f(i−1,j)+f(i−1,(j−1) mod k)

這道題我們轉移是用狀態轉移因為每次都是前一個i-1轉移過來的

不妨設當前狀態

1 2 3 4 5 6

a b c d e f

那麼我們開一個矩陣可以知道

1 2 3 4 5 6

1 1 1

2 1 1

3 1 1

4 1 1

5 1 1

6 1 1

我一開始初值給錯了，應該給0~k-1中每個ii都給1

因為我們知道這相當於給取i個為1留了一種可能

#include<cstdio>
#include<cstring>
#define N 55
struct matrix{
    int f[N][N],l,c;
}ans,m;
int n,p,k,r;
inline matrix multiply(matrix a,matrix b){
    matrix c;memset(c.f,0,sizeof(c.f));
    c.l=a.l;c.c=b.c;
    for
 
 (int i=0;i<a.l;++i)
        for (int j=0;j<b.c;++j)
            for (int z=0;z<a.c;++z){
                c.f[i][j]=((long long) a.f[i][z]*b.f[z][j]%p+c.f[i][j])%p;
            }
    return c;
}
int main(){
    freopen("3746.in","r",stdin);
    scanf("%d%d%d%d",&n,&p,&k,&r);long
 
 long t=(long long)n*k;
    m.c=m.l=k;
    for (int i=0;i<k;++i) m.f[i][i]++,m.f[(i-1+k)%k][i]++;
    ans.c=k;ans.l=k;
    for (int i=0;i<k;++i) ans.f[i][i]=1;
    for (;t;t>>=1,m=multiply(m,m)){
        if (t&1) ans=multiply(m,ans);
    }
    printf("%d",ans.f[0][r]);
    return 0;
}