bzoj3209 花神的数论题（数位dp+快速幂）

考虑有x个1的有s[x]个，那么答案就是$\prod x^{s[x]}$
那么我们现在的问题就是如何求<=n的数中有几个数的二进制表示有x个1.我们先处理出f[i][j]表示i位数有j个1的方案数，然后加上大小的限制，搞一搞就好了。我的写法是求比n小的有多少个，所以要给n++
复杂度$O(log^2n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define ll long long
#define mod 10000007
#define inf 0x3f3f3f3f
#define N 60
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline ll read(){
    ll x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
ll n,f[N][N],ans=1;//f[i][j]---i位数有j个1的方案数
int a[N],tot=-1;
inline ll ksm(ll x,ll k){
    ll res=1;for(;k;k>>=1,x=x*x%mod) if(k&1) res=res*x%mod;return res;
}
inline ll calc(int x){
    ll res=0;
    for(int i=tot;i>=0;--i){
        if(!a[i]) continue;
        res+=f[i][x];--x;
        if(x<0) break;
    }return res;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();ll x=n+1;
    while(x){
        a[++tot]=x&1;
        x>>=1;
    }for(int i=0;i<=tot;++i) f[i][0]=1;
    for(int i=1;i<=tot;++i)
        for(int j=1;j<=i;++j) f[i][j]=f[i-1][j]+f[i-1][j-1];
    for(int i=1;i<=tot+1;++i) (ans*=ksm(i,calc(i)))%=mod;
    printf("%lld\n",ans);
    return 0;
}