参考题解:点击打开链接
并不是很懂,尤其是出现了反例
不过这个dp求方案还是可以好好学下
dp[i][j]表示前(i-1)个二进制位用了j个位置的方案数,状态转移就是枚举当前的二进制位有多少个1
最后一步算Ans的组合是本身白棋和黑棋之间的距离是确定的,如果白棋位置确定后,那么黑棋也就确定,所以自由的位置就是n-j-K/2,
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N=10000+10;
const int M=1000000007;
int n,K,d;
int dp[20][N];
int c[N][200];
int C(int n,int m){
if (m>n-m)m=n-m;
return c[n][m];
}
void preC()
{
for (int i=0;i<N;i++)c[i][0]=1;
for (int i=1;i<N;i++)for (int j=1;j<=i && j<=100;j++)
c[i][j]=(c[i-1][j]+c[i-1][j-1])%M;
}
int main()
{
preC();
scanf("%d%d%d",&n,&K,&d);
dp[0][0]=1;
for (int i=0;i<15;i++)for (int j=0;j<=n-K;j++)for (int k=0;k*(d+1)<=K/2 && j+k*(d+1)*(1<<i)<=n-K;k++)
dp[i+1][j+k*(d+1)*(1<<i)]=(dp[i+1][j+k*(d+1)*(1<<i)]+1LL*dp[i][j]*C(K/2,k*(d+1))%M)%M;
int Ans=0;
for (int j=0;j<=n-K;j++)Ans=(Ans+1LL*dp[15][j]*C(n-j-K/2,K/2)%M)%M;
printf("%d\n",((C(n,K)-Ans)+M)%M);
return 0;
}