[BZOJ4870][Shoi2017]组合数问题 矩阵快速幂

诶感觉这题很套路啊
听说赛场上写出来的人不多
可能比赛的时候比较紧张这题放最后一题没那么好像吧
解法到不是特别难

相当于求一个模意义下的杨辉三角嘛
塞到矩阵里面转移就好了

#include <bits/stdc++.h>
#define N 55
using namespace std;
typedef long long LL;
inline int rd() {
    int x=0,f=1;char ch=getchar();
    while (ch>'9'||ch<'0') {if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int F[N],G[N],n,k,p,r;
inline void inc(int &x,int y) {x=(x+y)%p;}

inline void mul(int *R, int *A, int *B) {
    int ret[N];
    memset(ret,0,sizeof(ret));
    for (int i=0;i<k;i++)
        for (int j=0;j<k;j++)
            inc(ret[(i+j)%k], 1LL*A[i]*B[j]%p);
    for (int i=0;i<k;i++) R[i] = ret[i];
}

inline void qp(int *F,int *G,LL b) {
    int ret[N];
    memset(ret,0,sizeof(ret));
    ret[0] = 1;
    while (b) {
        if (b&1) mul(ret, ret, G);
        b >>= 1, mul(G, G, G);
    }
    for (int i=0;i<k;i++) F[i] = ret[i];
}

int main() {
    n = rd(), p = rd(), k = rd(), r = rd();
    G[0]++, G[1%k]++;
    qp(F, G, 1LL*n*k);
    printf("%d\n",(int)F[r]);
    return 0;
}