bzoj 3209 花神的数论题——二进制下的数位dp

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3209

可以枚举 “1的个数是...的数有多少个” ，然后就是用组合数算在多少位里选几个1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int mod=1e7+,N=;//(1e7+7)%941==0
ll n,dg[N];
int jc[N],jcn[N],cnt,ans=;
int pw(int x,int k)
{
    int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;
}
int gcd(int a,int b){return b?gcd(b,a%b):a;}
void exgcd(int a,int b,int &x,int &y)
{
    if(!b){x=;y=;return;}
    exgcd(b,a%b,y,x);y-=a/b*x;
}
void init()
{
    jc[]=;
    for(int i=;i<=;i++)jc[i]=(ll)jc[i-]*i%mod;
    int y;exgcd(jc[],mod,jcn[],y);
    for(int i=;i>=;i--)jcn[i]=(ll)jcn[i+]*(i+)%mod;
}
int C(int n,int m)
{
    if(m>n)return ;if(!m)return ;
    return (ll)jc[n]*jcn[m]%mod*jcn[n-m]%mod;
}
int main()
{
    init();
    scanf("%lld",&n);
    ll m=n;int p0=;
    while(m)
    {
        ll k=(m&-m);
        for(;(1ll<<p0)!=k;p0++);
        dg[++cnt]=p0;m-=(m&-m);
    }
    for(int i=;i<=dg[cnt]+;i++)
        for(int j=cnt,k=;j>=&&k<=i;j--,k++)
            ans=(ll)ans*pw(i,C(dg[j],i-k))%mod;//模数不是质数，不能对指数取模！！！
    printf("%d\n",ans);
    return ;
}

模数有毒吧怎么不是质数啊！这样都没法对指数取模了！

然后得知是数位dp。同样是枚举 “1的个数是...的数有多少个” ，但因为不是组合数，所以long long就行啦！

dp[ i ][ j ]表示第 i 位是0，后面从0..00到1..11中有多少个数有 j 个1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=,mod=1e7+;
ll n,dp[N+][N+];
int m,dg[N+],cnt,ans=;
void init()
{
    dp[][]=;
    for(int i=;i<=m;i++)
        for(int j=;j<=i;j++)
            dp[i][j]+=dp[i-][j]+dp[i-][j-];
}
int pw(int x,ll k)
{
    if(!x)return ;
    int ret=;while(k){if(k&1ll)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1ll;}return ret;
}
int main()
{
    scanf("%lld",&n);
    for(m=;m<=N&&(1ll<<m)<=n;m++);
    init();
    ll c=n;int p0=;
    while(c)
    {
        ll k=(c&-c);
        for(;(1ll<<p0)!=k;p0++);
        dg[++cnt]=p0+;c-=k;
    }
    for(int i=cnt,k=;i>=;i--,k++)
    {
        if(!i){ans=(ll)ans*k%mod;break;}
        for(int j=;j<=dg[i]-;j++)
            ans=(ll)ans*pw(j+k,dp[dg[i]][j])%mod;
    }
    printf("%d\n",ans);
    return ;
}