HDU 3555 Bomb(1-n含有“49”的数字个数)

题目链接：
HDU 3555 Bomb
题意：
求1到$n$中含有“49”序列的数字的个数。491，149都是含有“49”序列但是“419”不是的。
数据范围：$n\leq 2^{63} -1$
分析：
数位dp。差不多也是模版题了。
从高位到低位$dfs$记录前一位数字，并且记录当前是否已出现“49”序列，最后判断下状态，记忆化搜索。

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
using namespace std;
typedef long long ll;

int T;
ll n;
int digit[20];
ll dp[20][10][2][2];

ll dfs(int pos, int pre, int state, int limit)
{
    if (pos == -1) return state;
    if (dp[pos][pre][state][limit] != -1) return dp[pos][pre][state][limit];
    int last = limit ? digit[pos] : 9; // 是否已达数值上界
    ll ret = 0;
    for (int i = 0; i <= last; ++i) {
        ret += dfs(pos - 1, i, state || (i == 9 && pre == 4), limit && i == last);
    }
    dp[pos][pre][state][limit] = ret;
    return ret;
}

ll solve()
{
    int len = 0;
    memset(dp, -1, sizeof (dp));
    memset(digit, 0, sizeof (digit));
    while (n) {
        digit[len++] = n % 10;
        n /= 10;
    }
    return dfs(len - 1, 0, 0, 1);
}

int main()
{
    scanf("%d", &T);
    while (T--) {
        scanf("%lld", &n);
        printf("%lld\n", solve());
    }
    return 0;
}