bzoj3209 花神的数论题 数位DP

一开始想的是先预处理出n位的答案，因为假设当前给出的数x有y位，那么y-1位的答案是固定的，我只要处理y位的答案就可以了，问题是这样太复杂。。。
设f[i][j]表示i位有j个1的答案，那么明显有f[i][j]=f[i-1][j]+f[i-1][j-1]，表示第i位选0或1.
然后最后统计一下答案，就是每一种出现了多少次。。

其实挺简单的，基本想出来了，但是数位dp不熟，想的复杂了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
const int mo=10000007;
typedef long long ll;
ll n;
ll sum[100],f[100][100],g[100];
int wei[N];
int tot;
ll ans;
inline void pre()
{
    f[0][0]=1;
    fo(i,1,60)
    {
        f[i][0]=1;
        fo(j,1,i)
        f[i][j]=f[i-1][j-1]+f[i-1][j];
    }
}
ll solve(int x)
{
    ll sum=0;
    fd(i,tot,1)
    {
        if (wei[i]==1)
        {
            sum+=f[i-1][x];
            --x;
        }
        if (x<0)break;
    }
    return sum;
} 
ll pow(ll a,ll b)
{
    ll ret=1;
    while (b)
    {
        if (b&1)ret=ret*a%mo;
        a=a*a%mo;
        b>>=1;
    }
    return ret;
}
int main()
{
    pre();
    scanf("%lld",&n);
    ++n;
    tot=0;
    while (n)
    {
        wei[++tot]=n&1;
        n>>=1;
    }
    ans=1ll;
    fo(i,1,tot)
    ans=ans*pow(i,solve(i))%mo;
    printf("%lld\n",ans%mo);
    return 0;
}