Description

Solution

考慮這個式子的組合意義:
從 \(n*k\) 個球中取若幹個球,使得球的數量 \(\%k=r\) 的方案數
可以轉化為 \(DP\) 模型,設 \(f[i][j]\) 表示前 \(i\) 個步,取得球的數量 \(\%k=j\) 的方案數
\(f[i][j]=f[i-1][j]+f[i-1][j-1]\)
發現這個東西就是楊輝三角(胡話,此題無關)
這樣就可以做 \(O(k^3log)\) 了,並且可以過了

網上還有一種做法:
設 \(f[i*2][a+b]=\sum f[i][a]*f[i][b]\)
然後矩陣就變成了一個行向量了,復雜度優化成了 \(O(k^2log)\)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=51;
int mod,k,r;ll n;
struct mat{
    int a[N];
    mat(){memset(a,0,sizeof(a));}
    inline mat operator *(const mat &p){
        mat ret;
        for(int i=0;i<k;i++)
            for(int j=0;j<k;j++)
                ret.a[(i+j)%k]=(ret.a[(i+j)%k]+1ll
 
*a[i]*p.a[j])%mod;
        return ret;
    }
}S,T;
inline int qm(int x,int k){
    ll sum=1;
    while(k){
        if(k&1)sum=1ll*x*sum%mod;
        x=1ll*x*x%mod;k>>=1;
    }return sum;
}
int main(){
  freopen("pp.in","r",stdin);
  freopen("pp.out","w"
 
,stdout);
  cin>>n>>mod>>k>>r;
  if(k==1)printf("%d\n",qm(2,n)),exit(0);
  S.a[0]=1;S.a[1]=1;T=S;

  n=n*k-1;
  while(n){
      if(n&1)S=S*T;
      T=T*T;n>>=1;
  }
  printf("%d\n",S.a[r]);
  return 0;
}
    

bzoj 4870: [Shoi2017]組合數問題