HDU 3652 B-number (简单数位DP)

dp[i][s][k] k为模13的余数。

s=0，不存在；

s=1，不存在，上一位为1；

s=2，存在。

我的代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

int n,dp[12][15][3];
int bit[12],top;

int get_newk(int k,int d){
    if(k == 0){
        if(d == 1) return 1;
        else return 0;
    }
    if(k == 1){
        if(d == 1) return 1;
        if(d == 3) return 2;
        return 0;
    }
    return 2;
}

int dfs(int i,int s,int k,bool e){
    if(i == -1) return (s == 0 && k == 2) ? 1 : 0;
    if(!e && dp[i][s][k] != -1) return dp[i][s][k];
    int res = 0;
    int d,u = e ? bit[i] : 9;
    for(d = 0 ; d <= u ; d++){
        res += dfs(i-1,(s*10+d)%13,get_newk(k,d),e&&(d==u));
    }
    return e ? res : dp[i][s][k] = res;
}

int solve(int n){
    top = 0;
    for(;n;n/=10) bit[top++] = n%10;
    return dfs(top-1,0,0,1);
}

int main(){
    memset(dp,-1,sizeof(dp));
    while(~scanf("%d",&n)){
        printf("%d\n",solve(n));
    }

    return 0;
}