bzoj3209 花神的数论题（数位dp）

题目链接

题目描述：
设$sum(i)$表示$i$的二进制表示中1的个数
求$∏_{i=1}^{n}sum(i)$

分析：
$10^{15}<2^{50}$
显然sum的最大值为49
直接计算$sum(i)$比较麻烦，我们可以考虑计算有多少数的$sum值=X$（枚举$X$）
（有点反演的感觉）

一开始我设计了一个比较复杂的状态
$f[i][j][0/1][0/1]$：表示第$i$位，有$j$个1，这一位是0还是1，卡不卡上界

写完之后拍了拍，发现小数据还是可以的，大数据不知道为什么Wa掉了
（倒是可以用来拍一拍小数据）

if (n&(1LL<<_)) f[1][1][1][1]=1;
f[1][0][0][0]=1;
for (int i=1;i<_+1;i++)
    for (int j=0;j<=i;j++)
        for (int k=0;k<=1;k++)
            for (int s=0;s<=1;s++) 
            if (f[i][j][k][s]) {
                if (n&(1LL<<(_-i))) {        //1
                    if (s) {                 //卡上界 
                        f[i+1][j+1][1][1]+=f[i][j][k][s]; 
                        f[i+1][j][0][0]+=f[i][j][k][s]; 
                    } else {
                        f[i+1][j+1][1][0]+=f[i][j][k][s]; 
                        f[i+1][j][0][0]+=f[i][j][k][s]; 
                    }
                }
                else {                       //0 
                    if (s) {                 //卡上界 
                        f[i+1][j][0][1]+=f[i][j][k][s]; 
                    } else {
                        f[i+1][j+1][1][0]+=f[i][j][k][s]; 
                        f[i+1][j][0][0]+=f[i][j][k][s]; 
                    }
                }
            }

新状态：
$f[i][j]$表示$i$位长的数字，其中有$j$个1的数字个数
状态转移方程：$f[i][j]=f[i-1][j]+f[i-1][j-1]$
什么，这么简单？
$i$位长的数字，一定从$i-1$位的数字转移而来
而$i$位的数字转移有两种，即在$i-1$位的数字后加$0$或者加$1$
初始化：$f[1…n][0]=1$
（实际上就是在$j$个位置上选择$i$个位置放1）

#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long

using namespace std;

const ll p=1e7+7;
ll f[65][65],n;
int _,a[65];

ll KSM(ll a,ll b) {
    ll t=1;
    while (b) {
        if (b&1) t=(t*a)%p;
        a=(a*a)%p;
        b>>=1;
    }
    return t%p;
}

void prepare() {
    f[0][0]=1;
    for (int i=1;i<=60;i++) {
        f[i][0]=1;
        for (int j=1;j<=i;j++)
            f[i][j]=f[i-1][j]+f[i-1][j-1];
    }
}

ll solve(int x) {
    ll sum=0;
    for (int i=_;i>=1;i--) {
        if (a[i]==1) {
            sum+=f[i-1][x];
            --x;            //第i位固定为1 
        }
        if (x<0) break;
    }
    return sum;
}

int main() {
    prepare();
    scanf("%lld",&n);           
    n++;                    //计算小于n的个数
    _=0;
    while (n) {
        a[++_]=n&1;
        n>>=1;
    }
    ll ans=1;
    for (int i=1;i<=_;i++) 
        ans=(ans*KSM(i,solve(i))%p)%p;
    printf("%lld\n",ans);
    return 0;
}