bzoj3209 花神的数论题 数位dp

        网上题解都是高大上的组合数学。。然而我等蒟蒻显然只会简单粗暴的数位dp,即:

       令f[i][j][k]表示在前i位中,状态为j,有k个1的方案数。其中j=0表示前i为<n,j=1表示>=n。方便起见可以令n++,然后转移就很方便了,枚举一下01即可。

AC代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#define mod 10000007
#define ll long long
using namespace std;

ll n,dp[105][2][105];
int ksm(int x,ll y){
	int t=1; for (; y; y>>=1,x=(ll)x*x%mod) if (y&1) t=(ll)t*x%mod;
	return t;
}
int main(){
	scanf("%lld",&n); n++; int i,j,k,x;
	dp[1][1][0]=1;
	for (i=2; n; n>>=1,i++)
		for (j=0; j<2; j++)
			for (k=0; k<=50; k++)
				for (x=0; x<=max(j,(int)(n&1)) && x<=k; x++)
					dp[i][j][k]+=dp[i-1][j|(x<(n&1))][k-x];
	k=i-1; int ans=1;
	for (i=1; i<=50; i++) ans=(ll)ans*ksm(i,dp[k][0][i])%mod;
	printf("%d\n",ans);
	return 0;
}


by lych

2016.4.21

你可能感兴趣的:(动态规划,快速幂,数位dp)