dirty机房训练赛-chess

开始以为是矩阵快速幂之类的题

没想到居然是DP

不过其实也快想到了

不过关于最后剩下的几排出现了误区

他们实际上也应该与前面一样的

转移十分合理到位

也就不多说了

这是我的优美代码

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
int n,C,fac[50005],inv[50005],another,ans,dp[105][10005],f[105][105];
long long m;
int c(int x,int y)
{
	if(y>x) return 0;
	return 1ll*fac[x]*inv[y]%mod*inv[x-y]%mod;
}
int moc(int x)
{
	if(x>=mod) return x-mod;
	return x;
}
int fast(int x,long long y)
{
	int ans=1;
	while(y)
	{
		if(y&1)
		{
			ans=1ll*ans*x%mod;
		}
		y>>=1;
		x=1ll*x*x%mod;
	}
	return ans;
}
int main()
{
	freopen("chess.in","r",stdin);
	freopen("chess.out","w",stdout);
	cin>>n>>m>>C;
	fac[0]=inv[0]=inv[1]=1;
	for(int i=1;i<=50000;i++)
	fac[i]=1ll*fac[i-1]*i%mod;
	for(int i=2;i<=50000;i++)
	inv[i]=1ll*inv[mod%i]*(mod-mod/i)%mod;
	for(int i=2;i<=50000;i++)
	inv[i]=1ll*inv[i-1]*inv[i]%mod;
	dp[0][0]=1;
	for(int i=0;i<=n;i++)
	{
		for(int j=0;j<=n;j++)
		{
			f[i][j]=fast(c(n,i),(m-j)/n+1);
		}
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<=C&&j<=i*n;j++)
		{
			for(int k=0;k<=j&&k<=n;k++)
			{
				dp[i][j]=moc(dp[i][j]+1ll*dp[i-1][j-k]*f[k][i]%mod);
			}
		}
	}
	cout<<dp[n][C];
	return 0;
}