首页 > 代码库 > 花神的数论题(数位dp)

花神的数论题(数位dp)

规定sum[i] 为i里面含1的个数 ,求从1-N sum[i]的乘积。

数为64位内的,也就是sum[i]<=64的,这样可以dp求出1-N中含k个1的数有多少个,快速幂一下就可以了。

有个地方没开LL ,WA了几次。

 1 #include <iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #include<stdlib.h> 6 #include<vector> 7 #include<cmath> 8 #include<queue> 9 #include<set>10 using namespace std;11 #define LL long long12 #define mod 1000000713 #define INF 0xfffffff14 const double eps = 1e-8;15 const double pi = acos(-1.0);16 const double inf = ~0u>>2;17 LL dp[65][65][65];18 int di[65];19 LL dfs(int i,bool e,int k,int s)20 {21     if(i==-1) return k==s;22     if(!e&&~dp[i][k][s]) return dp[i][k][s];23     int mk = e?di[i]:1;24     LL ans = 0;25     for(int j = 0 ; j <= mk ; j++)26     {27         ans = ans+dfs(i-1,e&&(j==mk),k,s+j);28     }29     return e?ans:dp[i][k][s] = ans;30 }31 LL exp_mod(int a,LL n)32 {33     LL t;34     if(n==0) return 1;35     if(n==1) return a;36     t = exp_mod(a,n/2);37     t = (t*t)%mod;38     if(n&1) t = (t*a)%mod;39     return t;40 }41 LL cal(LL n)42 {43     int g=0;44     while(n)45     {46         di[g++] = n%2;47         n/=2;48     }49     LL ans = 1;50     for(int i = 1 ;i <= g ; i++)51     {52         LL num = dfs(g-1,1,i,0);53         ans = (ans*exp_mod(i,num))%mod;54     }55     return ans;56 }57 int main()58 {59     LL n;60     memset(dp,-1,sizeof(dp));61     while(cin>>n)62     {63         cout<<cal(n)<<endl;64     }65     return 0;66 }
View Code