HDU 4507 吉哥系列故事——恨7不成妻(数位DP)
2015-07-25 14:25
489 查看
Description
如果一个整数符合下面3个条件之一,那么我们就说这个整数和7有关
1、整数中某一位是7;
2、整数的每一位加起来的和是7的整数倍;
3、这个整数是7的整数倍;
现在问一定区间内和7无关的数字的平方和
Input
输入数据的第一行是case数T(1 <= T <= 50),然后接下来的T行表示T个case;每个case在一行内包含两个正整数L, R(1 <= L <= R <= 10^18)。
Output
请计算[L,R]中和7无关的数字的平方和,并将结果对10^9 + 7 求模后输出。
Sample Input
3
1 9
10 11
17 17
Sample Output
236
221
0
Solution
数位DP,用结构数组dp[cur][p1][p2]{ll cnt,ll sum,ll sqsum}表示当当前位为cur位,前面那些数位和模7值为p1,前面那些数位构成的数模7值为p2这个状态时,前面所有与7无关的数的数量为cnt,其和为sum,平方和为sqsum,以p[i]表示10^i,则当前面位确定,当前位为i,状态为结构ans,搜索出后面数位的状态为结构temp时
ans.cnt+=temp.cnt
ans.sum+=i*p[cur]*temp.cnt+temp.sum
ans.sqsum+=i*i*p[cur]*p[cur]*temp.cnt+2*i*p[cur]*temp.sum+temp.sqsum
依次递推,终止条件为当前位为最后一位,终止状态为最后一位是否与7无关
Code
如果一个整数符合下面3个条件之一,那么我们就说这个整数和7有关
1、整数中某一位是7;
2、整数的每一位加起来的和是7的整数倍;
3、这个整数是7的整数倍;
现在问一定区间内和7无关的数字的平方和
Input
输入数据的第一行是case数T(1 <= T <= 50),然后接下来的T行表示T个case;每个case在一行内包含两个正整数L, R(1 <= L <= R <= 10^18)。
Output
请计算[L,R]中和7无关的数字的平方和,并将结果对10^9 + 7 求模后输出。
Sample Input
3
1 9
10 11
17 17
Sample Output
236
221
0
Solution
数位DP,用结构数组dp[cur][p1][p2]{ll cnt,ll sum,ll sqsum}表示当当前位为cur位,前面那些数位和模7值为p1,前面那些数位构成的数模7值为p2这个状态时,前面所有与7无关的数的数量为cnt,其和为sum,平方和为sqsum,以p[i]表示10^i,则当前面位确定,当前位为i,状态为结构ans,搜索出后面数位的状态为结构temp时
ans.cnt+=temp.cnt
ans.sum+=i*p[cur]*temp.cnt+temp.sum
ans.sqsum+=i*i*p[cur]*p[cur]*temp.cnt+2*i*p[cur]*temp.sum+temp.sqsum
依次递推,终止条件为当前位为最后一位,终止状态为最后一位是否与7无关
Code
#include<cstdio> #include<cstring> #include<iostream> using namespace std; typedef long long ll; const ll mod=1000000007ll; struct node { ll cnt;//和7无关的数的个数 ll sum;//这些数的和 ll sqsum;//这些数的平方和 }dp[20][10][10];//第一位存第几位,第二维存到该位所有位数和模7,第三位存到该位时数值模7 int num[20]; ll p[20]; node dfs(int cur,int p1,int p2,bool fp) { if(cur==-1) { node temp; temp.cnt=(p1!=0&&p2!=0); temp.sum=temp.sqsum=0; return temp; } if(!fp&&dp[cur][p1][p2].cnt!=-1) return dp[cur][p1][p2]; int fpmax=fp?num[cur]:9; node ans,temp; ans.cnt=ans.sum=ans.sqsum=0; for(int i=0;i<=fpmax;i++) { if(i==7)continue; temp=dfs(cur-1,(p1+i)%7,(p2*10+i)%7,fp&&i==fpmax); ans.cnt+=temp.cnt; ans.cnt%=mod; ans.sum+=(temp.sum+((p[cur]*i)%mod)*temp.cnt%mod)%mod; ans.sum%=mod; ans.sqsum+=(temp.sqsum+((2*p[cur]*i)%mod)*temp.sum)%mod; ans.sqsum%=mod; ans.sqsum+=(temp.cnt*p[cur])%mod*p[cur]%mod*i*i%mod; ans.sqsum%=mod; } if(!fp) dp[cur][p1][p2]=ans; return ans; } ll f(ll n) { int len=0; while(n) { num[len++]=n%10; n/=10; } return dfs(len-1,0,0,1).sqsum;//输出的是所有与7无关的数的平方和 } int main() { p[0]=1;//打表 for(int i=1;i<20;i++) p[i]=(p[i-1]*10)%mod; for(int i=0;i<20;i++)//初始化 for(int j=0;j<10;j++) for(int k=0;k<10;k++) dp[i][j][k].cnt=-1; int t; scanf("%d",&t); while(t--) { ll a,b; scanf("%lld%lld",&a,&b); ll ans=f(b); ans-=f(a-1); ans=(ans%mod+mod)%mod;//注意大数做减法的取模 printf("%lld\n",ans); } return 0; }
相关文章推荐
- UVa 11134 - Fabled Rooks(贪心)
- poj 2570 Fiber Network(floyd)
- 黑马程序员——集合框架(Map详解)
- 从源代码分析modelDriven拦截器和params拦截器和拦截器prepare 和paramsPrepareParamsStack拦截器栈(让你的Struts2代码更简洁——如何培养框架设计能力
- poj 2570 Fiber Network(floyd)
- [黑马程序员](第16-22天)IO流(下)
- iOS __weak学习碰到的疑问
- POJ 2251 Dungeon Master
- ProgressDialog使用总结
- JBOSS ESB企业服务器总线
- [黑马程序员](第16-22天)IO流(上)
- [黑马程序员](第16-22天)IO流(中)
- myeclipse 2015 安装 aptana3.6
- 教你如何解包、打包boot.img以及获取adb root权限
- 每天一个Linux命令-9(Find–atime –ctime –mtime)
- 卡特兰数
- android studio 运行代码时device不能识别genymotion设备
- JAVA环境变量配置
- http://blog.csdn.net/ooflywing/article/details/5616311
- 在caffe上跑自己的数据