BZOJ_3209_花神的数论题_组合数+数位DP

BZOJ_3209_花神的数论题_组合数+数位DP

Description

背景
众所周知，花神多年来凭借无边的神力狂虐各大 OJ、OI、CF、TC …… 当然也包括 CH 啦。
描述
话说花神这天又来讲课了。课后照例有超级难的神题啦…… 我等蒟蒻又遭殃了。
花神的题目是这样的
设 sum(i) 表示 i 的二进制表示中 1 的个数。给出一个正整数 N ，花神要问你
派(Sum(i)),也就是 sum(1)—sum(N) 的乘积。

Input

一个正整数 N。

Output

一个数，答案模 10000007 的值。

Sample Input

样例输入一

3

Sample Output

样例输出一

2

HINT

对于样例一，1*1*2=2；

数据范围与约定

对于 100% 的数据，N≤10^15

---

设f[i][j]表示所有i位数中1的个数为j的数的个数。

然后发现这是组合数，相当于i个数中选j个。

然后数位DP求出每个数在答案中出现了多少次。

乘起来即可。

代码:

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll mod=10000007;
ll c[150][150],n,cnt[150];
ll qp(ll x,ll y) {
    ll re=1;for(;y;y>>=1ll,x=x*x%mod) if(y&1ll) re=re*x%mod; return re;
}
int main() {
    int i,j;
    for(i=0;i<=60;i++) c[i][0]=c[i][i]=1;
    for(i=1;i<=60;i++) {
        for(j=1;j<i;j++) {
            c[i][j]=c[i-1][j]+c[i-1][j-1];
        }
    }
    scanf("%lld",&n); n++;
    int now=0;
    for(i=60;i>=1;i--) {
        // printf("%d",(n&(1ll<<(i-1)))!=0);
        if(!(n&(1ll<<(i-1)))) continue;
        for(j=1;j<=60;j++) {
            if(j>=now&&j-now<=i-1) cnt[j]+=c[i-1][j-now];
        }
        now++;
    }
    // puts("");
    ll ans=1;
    for(i=1;i<=60;i++) {
        // printf("%lld
",cnt[i]);
        ans=ans*qp(i,cnt[i])%mod;
    }
    printf("%lld
",ans);
}