hdu3555 Bomb 数位dp

//3555

//题意：统计 1 到 n 中包含 49 的数的个数

#include "bitsstdc++.h"
using namespace std;
__int64 dp[22][3];

int main()
{
    int i;
    dp[0][0] = 1;
    for(i = 1; i <= 20; ++i) {
        dp[i][0] = dp[i - 1][0] * 10 - dp[i - 1][1] * 1;
        dp[i][1] = dp[i - 1][0] * 1;
        dp[i][2] = dp[i - 1][1] * 1 + dp[i - 1][2] * 10;
    }
    int T;
    scanf("%d", &T);
    while(T--) {
        __int64 n;
        scanf("%I64d", &n);
        int len = 0;
        int digit[22];
        memset(digit, 0, sizeof(digit));
        ++n;
        while(n) {
            ++len;
            digit[len] = n % 10;
            n /= 10;
        }
        __int64 res = 0;
        bool _49 = 0;
        for(i = len; i >= 1; --i) {
            //如果已经出现 49 ，则加上第 i 位 -1 后所有数的个数
            if(_49) {
                res += (dp[i - 1][0] + dp[i - 1][2]) * digit[i];
            }
            //如果未出现 49， 则加上第 i 位 -1 后所有符合要求的数的个数
            else {
                res += dp[i - 1][2] * digit[i];
                if(digit[i] > 4)
                    res += dp[i - 1][1] * 1;
                if(digit[i + 1] == 4 && digit[i] == 9)
                    _49 = 1;
            }
            //下一次循环处理，第 i 位 == 原数的情况
        }
        printf("%I64d
", res);
    }
}