【BZOJ3209】花神的数论题 数位DP

【BZOJ3209】花神的数论题

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

题解：又一个题目名称和题本身一点关系都没有的~

很容易想到按位拆分，分别考虑1的个数是k的数有多少个，然后快速幂一下计算贡献

怎么知道1的个数是k的数有多少个呢？预处理出组合数，然后数位DP吧！（对本蒟蒻来说就是INF的细节）

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
typedef long long ll;
const ll mod=10000007;
ll c[60][60];
ll cnt[60];
ll n,sum,ans;
ll pm(ll x,ll y)
{
	ll z=1;
	while(y)
	{
		if(y&1)	z=z*x%mod;
		x=x*x%mod,y>>=1;
	}
	return z;
}
int main()
{
	c[0][0]=1;
	ll i,j;
	for(i=1;i<=50;i++)
	{
		c[i][0]=1;
		for(j=1;j<=i;j++)	c[i][j]=c[i-1][j-1]+c[i-1][j];
	}
	scanf("%lld",&n);
	for(i=50;i;i--)
	{
		if(n&(1ll<<i-1))
		{
			for(j=sum;j<=50;j++)	cnt[j]+=c[i-1][j-sum];
			sum++;
		}
	}
	cnt[sum]++;
	for(ans=i=1;i<=50;i++)	ans=ans*pm(i,cnt[i])%mod;
	printf("%lld",ans);
	return 0;
}