bzoj3209: 花神的数论题

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3209

题解

　　这种题。。。没有1A是耻辱，一开始模数打错了，以为是$10^9+7$，其实题目里是$10^7+7$
　　f[i][j]、g[i][j]分别表示在无、有限制的情况下i位数包含j个1的方案数。（允许前导零）
　　不难发现$log_210^{15}≈50$，因此题目中$sum(i)$不超过50，所以可以枚举$sum(i)$的值，题目就变成求$[1,N]$里有多少个数的二进制表示中恰好有x个1。
　　最后快速幂即可。

代码

//数位dp
#include <cstdio>
#include <algorithm>
#define maxn 60
#define ll long long
#define p 10000007
using namespace std;
ll f[maxn][maxn], g[maxn][maxn], num[maxn], N;
void init(ll x)
{for(;x;x>>=1)num[++N]=x bitand 1;}
void dp()
{
    ll i, j;
    for(i=0;i<=N;i++)f[i][0]=g[i][0]=1;
    for(i=1;i<=N;i++)for(j=1;j<=N;j++)f[i][j]=f[i-1][j-1]+f[i-1][j];
    g[0][0]=1;
    for(i=1;i<=N;i++)for(j=1;j<=N;j++)
        if(num[i]==1)g[i][j]=f[i-1][j]+g[i-1][j-1];
        else g[i][j]=g[i-1][j];
}
ll count(ll x)
{
    ll ans=g[N-1][x-1], i;
    for(i=N-1;i>=1;i--)ans+=f[i-1][x-1];
    return ans;
}
ll pow(ll a, ll b)
{
    ll t, ans;
    for(ans=1,t=a;b;t=(t*t)%p,b>>=1)if(b&1)ans=(ans*t)%p;
    return ans;
}
int main()
{
    ll n, i, ans=1;
    scanf("%lld",&n);
    init(n);
    dp();
    for(i=1;i<=N;i++)ans=(ans*pow(i,count(i)))%p;
    printf("%lld",ans);
    return 0;
}