HDU 3652 数位DP(递归)

递归

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
int a[11];
int n;
/*
* 0 包含13
* 1 不包含13，但以3开头
* 2 不包含13，不以3开头
*/
int dp[11][3][15];
int dfs(int len, int pre, int mod, bool lim, bool has) {
if(!len) return has && !mod;
if(!lim) {
if(!has && pre != 1 && ~dp[len][0][mod]) return dp[len][0][mod];
if(!has && pre == 1 && ~dp[len][1][mod]) return dp[len][1][mod];
if(has && ~dp[len][2][mod]) return dp[len][2][mod];
}
int m = lim ? a[len-1] : 9;
int ret = 0;
for(int i = 0; i <= m; i++) {
ret += dfs(len-1, i, (mod*10+i)%13, lim&&(i==m), has||(pre == 1 &&i==3));
}
if(!lim){
if(has) dp[len][2][mod] = ret;
if(!has && pre != 1) dp[len][0][mod] = ret;
if(!has && pre == 1) dp[len][1][mod] = ret;
}
return ret;
}
int gao(int n) {
int num = 0;
while(n) {
a[num++] = n % 10;
n /= 10;
}
memset(dp, -1, sizeof(dp));
return dfs(num, 0, 0, 1, 0);
}
int main() {
while(cin >> n) cout << gao(n) << endl;
return 0;
}