B-number

B-number

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=3652

数位dp

这题是暑期集训的时候做的，昨天补了数位dp的记忆化搜索做法，把艾神的递推算法更新一下。

代码如下：

#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#define LL long long
#define LEN 20
#define is_1 2
#define have_13 2
#define mod_by_13 13
#define pre_same 2
using namespace std;
LL t[LEN];
LL dp[LEN][is_1][have_13][mod_by_13][pre_same];
string s;
LL len;
void init_t();
void solve();
int main(void){
    init_t();
    while(cin>>s){
        len=s.size();
        memset(dp,0,sizeof(dp));
        dp[0][0][0][0][1]=1;
        solve();
        LL temp=0;
        for(int i=0;i<is_1;++i)
        for(int j=0;j<pre_same;++j)
            temp+=dp[len][i][1][0][j];
        cout<<temp<<endl;
    }
}
void init_t(){
    t[0]=1;
    for(int i=1;i<LEN;++i)
        t[i]=(t[i-1]*10)%13;
}
void solve(){
    for(int a=0;a<len;++a)
    for(int b=0;b<is_1;++b)
    for(int c=0;c<have_13;++c)
    for(int d=0;d<mod_by_13;++d)
    for(int e=0;e<pre_same;++e)
    if(dp[a][b][c][d][e]){
        int r=(e?s[a]-'0':9);
        for(int x=0;x<=r;++x){
            dp[a+1][x==1][c|(b&&x==3)][(d+t[len-a-1]*x)%13][e&&x==r]
                +=dp[a][b][c][d][e];
        }
    }
}

 感觉还是太懒了，这两周金工实习这么好的机会，居然每天也只能写一道题，而且是水题。