hdu4507
数位dp,终于守得云开见月明了。建议初学者先试试两道比较简单的hdu2089,hdu3555。
鸣谢:http://blog.csdn.net/acm_cxlove/article/details/8707084。
数位dp也是一种基于状态压缩、优化的动态规划。不同的是,它的压缩和优化往往基于数的一些特性。而数最基本的表现形式:a/b --- [a/b]、[a%b]。
这种dp才是体现一个人智慧的地方。(额外想为ACM竞赛的同学说两句,个人还是特别顶复旦出题的,至少它出的绝大部分题目都是可以自己通过大学以前的数学知识慢慢想到,而不是像某些学校的题目,只要听说过、学过、看过,某一个不知道哪里冒出来的数学理论就能瞬间AC,否则***。还是不放水了,言归正传)
本题,中文描述的,不解释了。初学者往往都会往容斥的方面想,(也不是不可以)但其实条件2、条件3的综合比较困难了,代码量和复杂度可能都会很大。
所以直接递归找跟7有关的数比较科学,假设原数字a+b位,如果搜索前a位fa(fa后b位都是0),后面b位跟7有关的任何一个数x,产生的结果和是(fa+x)^2+x^2。对所有的x,就有{fa^2*cnt(x)+2fa*sum(x)+sum(x^2)=f(a+b,b)}+f(b,0)。这个很容易想到,那么如果搜索时刚好按a从小到大拆分呢?不就是很明显了吗?
1 #include <iostream> 2 #include <cstring> 3 #include <cstdio> 4 #include <algorithm> 5 using namespace std; 6 #define LL long long 7 const LL mod = 1000000007LL; 8 #define mp(a,b) make_pair(a,b) 9 int bit[21]; 10 //长度,是否有7,数字和%7,数字%; 数字和、结果和、数字个数 11 long long s1[20][2][7][7],s2[20][2][7][7],cnt[20][2][7][7]; 12 LL fac[20]={1}; 13 typedef pair<pair<LL,LL>,LL> pll; 14 pll DP(int len,int a,int b,int c,int g){ 15 //printf("len = %d, %d, %d, %d, g = %d\n",len,a,b,c,g); 16 if(g && cnt[len][a][b][c] >= 0) 17 return mp(mp(cnt[len][a][b][c],s1[len][a][b][c]),s2[len][a][b][c]); 18 if(len <= 0){ 19 if(b&&c&&!a) cnt[0][a][b][c]=0; 20 else cnt[0][a][b][c]=1; 21 s1[0][a][b][c]=s2[0][a][b][c]=0; 22 return mp(mp(cnt[len][a][b][c],s1[len][a][b][c]),s2[len][a][b][c]); 23 } 24 int bound=bit[len]; if(g) bound=9; 25 LL tcnt=0,ts1=0,ts2=0; 26 int nl=len-1,na,nb,nc; 27 for(int i=0;i<=bound;i++){ 28 na=(a||i==7); nb=(b+i)%7; nc=(c*10+i)%7; 29 pll p=DP(nl,na,nb,nc,g||(i<bound)); 30 LL f = fac[nl]*i % mod; //f按位拆分,不用靠递归记录!!! 31 tcnt= (tcnt+p.first.first)%mod; 32 ts1 = (ts1+p.first.second+p.first.first*f)%mod; 33 ts2 = (ts2+p.first.first*(f*f%mod)%mod+p.first.second*f*2%mod+p.second)%mod; 34 } 35 if(g){ 36 cnt[len][a][b][c] = tcnt; 37 s1[len][a][b][c] = ts1; 38 s2[len][a][b][c] = ts2; 39 } 40 return mp(mp(tcnt,ts1),ts2); 41 } 42 LL sum(LL n){ 43 LL a=n,b=n+1,c=2*n+1; 44 LL x=3,y=2; 45 if(a%x==0) a/=x,x=1;if(a%y==0) a/=y,y=1; 46 if(b%x==0) b/=x,x=1;if(b%y==0) b/=y,y=1; 47 if(c%x==0) c/=x,x=1;if(c%y==0) c/=y,y=1; 48 a%=mod;b%=mod;c%=mod; 49 return (a*b%mod)*c%mod; 50 } 51 LL solve(LL n){ 52 if(n <= 0) return 0; 53 int len=0; 54 memset(bit,0,sizeof(bit)); 55 LL m=n; 56 while(n > 0) 57 bit[++len]=n%10, n/=10; 58 //cout<<"n = "<<m<<" ,len = "<<len<<endl; 59 return ((sum(m)-DP(len,0,0,0,0).second)%mod+mod)%mod; 60 } 61 int main() 62 { 63 for(int i=1;i<20;i++) 64 fac[i]=(fac[i-1]*10)%mod; 65 memset(cnt,-1,sizeof(cnt)); 66 int cases; cin>>cases; 67 for(int cas=1;cas<=cases;cas++){ 68 LL l,r; 69 cin>>l>>r; 70 //scanf("%lld%lld",&l,&r); 71 cout<<((solve(r)-solve(l-1))%mod+mod)%mod<<endl; 72 } 73 return 0; 74 }