首页 > 代码库 > 花神的数论题(数位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 }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。