B-number
B-number
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3652
数位dp
这题是暑期集训的时候做的,昨天补了数位dp的记忆化搜索做法,把艾神的递推算法更新一下。
代码如下:
1 #include<cstdio> 2 #include<string> 3 #include<cstring> 4 #include<iostream> 5 #define LL long long 6 #define LEN 20 7 #define is_1 2 8 #define have_13 2 9 #define mod_by_13 13 10 #define pre_same 2 11 using namespace std; 12 LL t[LEN]; 13 LL dp[LEN][is_1][have_13][mod_by_13][pre_same]; 14 string s; 15 LL len; 16 void init_t(); 17 void solve(); 18 int main(void){ 19 init_t(); 20 while(cin>>s){ 21 len=s.size(); 22 memset(dp,0,sizeof(dp)); 23 dp[0][0][0][0][1]=1; 24 solve(); 25 LL temp=0; 26 for(int i=0;i<is_1;++i) 27 for(int j=0;j<pre_same;++j) 28 temp+=dp[len][i][1][0][j]; 29 cout<<temp<<endl; 30 } 31 } 32 void init_t(){ 33 t[0]=1; 34 for(int i=1;i<LEN;++i) 35 t[i]=(t[i-1]*10)%13; 36 } 37 void solve(){ 38 for(int a=0;a<len;++a) 39 for(int b=0;b<is_1;++b) 40 for(int c=0;c<have_13;++c) 41 for(int d=0;d<mod_by_13;++d) 42 for(int e=0;e<pre_same;++e) 43 if(dp[a][b][c][d][e]){ 44 int r=(e?s[a]-'0':9); 45 for(int x=0;x<=r;++x){ 46 dp[a+1][x==1][c|(b&&x==3)][(d+t[len-a-1]*x)%13][e&&x==r] 47 +=dp[a][b][c][d][e]; 48 } 49 } 50 }
感觉还是太懒了,这两周金工实习这么好的机会,居然每天也只能写一道题,而且是水题。