CodeForces 232 B.Table（dp+组合数学）

Description

给出一个$n*m$的棋盘，在上面放一些棋子使得任意一个$n*n$的区域里都有$k$个棋子，问方案数

Input

三个整数$n,m,k(1\leq n\leq100,n\leq m\leq 10^{18},0\leq k\leq n^2)$

Output

输出方案数，结果模$10^9+7$

Sample Input

5 6 1

Sample Output

45

Solution

注意到前$n$列每列棋子数固定后，之后$m-n$列每列棋子数全部固定（第$i$列棋子个数等于第$i-n$列棋子个数）

$dp[i][j]$表示在前$i$列放$j$个棋子的合法方案数

考虑第$i$列放的棋子个数$k$，那么第$i+t\cdot n$棋子数均为$k$，$t=0,1,...,\lfloor\frac{m-i}{n}\rfloor$

一列放$k$个棋子方案数为$C_{n}^{k}$，$\lceil\frac{m-i}{n}\rceil$列方案数即为$(C_{n}^{k})^{\lceil\frac{m-i}{n}\rceil}$，记$A[i][k]=(C_{n}^{k})^{\lceil\frac{m-i}{n}\rceil}$

进而有转移方程：$dp[i][j]=\sum\limits_{k} dp[i-1][k]\cdot A[i][j-k],k=max(j-n,0),...,j$

答案即为$dp[n][k]$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define mod 1000000007
int n,k;
ll m,C[111][111],A[111][111],dp[111][11111];
ll mod_pow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
int main()
{
    scanf("%d%I64d%d",&n,&m,&k);
    C[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        C[i][0]=C[i][i]=1;
        for(int j=1;j<i;j++)C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
    }
    for(int i=1;i<=n;i++)
        for(int j=0;j<=n;j++)
            A[i][j]=mod_pow(C[n][j],(m-i+n)/n);
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<=i*n;j++)
            for(int k=max(j-n,0);k<=j;k++)
                dp[i][j]=(dp[i][j]+dp[i-1][k]*A[i][j-k]%mod)%mod;
    printf("%I64d\n",dp[n][k]);
    return 0;
}