hdu 3709+hdu 3555(数位dp)
题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3709
数位dp...完全看大牛模版理解的。。。。
View Code
1 #include<iostream> 2 using namespace std; 3 __int64 dp[19][19][2000]; 4 int digit[19]; 5 6 //pos表示当前的位置,o表示支点，pre表示从最高位到pos的力矩之和，doing表示是否有上限，若无，则为9; 7 __int64 dfs(int pos,int o,int pre,bool doing){ 8 if(pos==-1){ 9 return pre==0; //已经全部组合 10 } 11 if(pre<0)return 0;//如果前面组合的力矩之和已经为负，则后面的必然小于0； 12 //没有上限，且前面的状态已经搜过 13 if(!doing&&dp[pos][o][pre]!=-1){ 14 return dp[pos][o][pre]; 15 } 16 __int64 ans=0; 17 int end=doing?digit[pos]:9; 18 for(int i=0;i<=end;i++){ 19 int npre=pre; //枚举下一个状态 20 npre+=(pos-o)*i; 21 ans+=dfs(pos-1,o,npre,doing&&i==end); 22 } 23 if(!doing){ 24 dp[pos][o][pre]=ans; 25 } 26 return ans; 27 } 28 29 __int64 solve(__int64 n){ 30 int pos=0; 31 while(n){ 32 digit[pos++]=n%10; 33 n/=10; 34 } 35 __int64 ans=0; 36 for(int o=0;o<pos;o++){ 37 ans+=dfs(pos-1,o,0,1); 38 } 39 return ans-(pos-1);//0,00,000....这种情况 40 } 41 42 int main(){ 43 int _case; 44 scanf("%d",&_case); 45 memset(dp,-1,sizeof(dp)); 46 while(_case--){ 47 __int64 n,m; 48 scanf("%I64d%I64d",&n,&m); 49 printf("%I64d\n",solve(m)-solve(n-1)); 50 } 51 return 0; 52 }
题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3555
View Code
1 #include<iostream> 2 using namespace std; 3 __int64 dp[20][3]; 4 int digit[20]; 5 6 //have==1表示前一位为4,have==0表示没有出现49，have==2表示前面有49出现 7 __int64 dfs(int pos,int have,bool doing){ 8 if(pos==-1){ 9 return have==2;//找到一个解 10 } 11 if(!doing&&dp[pos][have]!=-1){ 12 return dp[pos][have]; 13 } 14 __int64 ans=0; 15 int end=doing?digit[pos]:9; 16 for(int i=0;i<=end;i++){ 17 int nhave=have; 18 if(have==1&&i!=4){ 19 nhave=0; 20 } 21 if(have==0&&i==4){ 22 nhave=1; 23 } 24 if(have==1&&i==9){ 25 nhave=2; 26 } 27 ans+=dfs(pos-1,nhave,doing&&i==end); 28 } 29 if(!doing) 30 dp[pos][have]=ans; 31 return ans; 32 } 33 34 __int64 solve(__int64 n){ 35 int pos=0; 36 while(n){ 37 digit[pos++]=n%10; 38 n/=10; 39 } 40 return dfs(pos-1,0,1); 41 } 42 43 int main(){ 44 int _case; 45 scanf("%d",&_case); 46 memset(dp,-1,sizeof(dp)); 47 while(_case--){ 48 __int64 n; 49 scanf("%I64d",&n); 50 printf("%I64d\n",solve(n)); 51 } 52 return 0; 53 }
昨天做的fzu的月赛。。。当时没过。。。
http://acm.fzu.edu.cn/contest/problem.php?cid=126&sortid=1
这个代码我没提交过。。。
View Code
1 #include<iostream> 2 #include<cstring> 3 using namespace std; 4 __int64 dp[20][20]; 5 int digit[20]; 6 7 __int64 DFS(int pos,int have,bool doing){ 8 if(pos==-1){ 9 return have; 10 } 11 if(!doing&&dp[pos][have]!=-1){ 12 return dp[pos][have]; 13 } 14 __int64 ans=0; 15 int end=doing?digit[pos]:9; 16 for(int i=0;i<=end;i++){ 17 int nhave=have; 18 if(i==1)nhave=have+1; 19 ans+=DFS(pos-1,nhave,doing&&i==end); 20 } 21 if(!doing){ 22 dp[pos][have]=ans; 23 } 24 return ans; 25 } 26 27 28 __int64 solve(__int64 n){ 29 int pos=0; 30 while(n){ 31 digit[pos++]=n%10; 32 n/=10; 33 } 34 return DFS(pos-1,0,1); 35 } 36 37 int main(){ 38 __int64 n,m; 39 memset(dp,-1,sizeof(dp)); 40 while(~scanf("%I64d%I64d",&n,&m)){ 41 printf("%I64d\n",solve(m)-solve(n-1)); 42 } 43 return 0; 44 } 45 46
PS:记忆化搜索真的很强大啊。。。
