HDU 3555 Bomb（数位dp）

题目链接：【HDU 3555】

求出1~n(1 <= N <= 2^63-1)数字包含49的有多少个

神奇的数位dp

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
#define ll __int64
int t, arr[25];
ll dp[25][3];
/*
dp[i][0]:位数为i，没有49 
dp[i][1]:9开头的，没有49 
dp[i][2]:有49 
*/ 
void init()
{
	dp[0][0]=1;
	for(int i=1; i<=22; i++)
	{
		dp[i][0] = dp[i-1][0]*10-dp[i-1][1];
		dp[i][1] = dp[i-1][0];
		dp[i][2] = dp[i-1][2]*10+dp[i-1][1];
	}
}
ll solve(ll x)
{
	int len = 0;
	while(x)
	{
		arr[++len] = x%10;
		x/=10;
	}
	arr[len+1] = 0;
	ll ans=0, flag=0;
	for(int i=len; i>=1; i--)
	{
		ans += dp[i-1][2]*arr[i];
		if(flag) ans+=dp[i-1][0]*arr[i];
		if(!flag && arr[i]>4) ans+=dp[i-1][1];
		if(arr[i+1]==4 && arr[i]==9) flag=1;
	}
	return ans;
}
int main()
{
	init();
	scanf("%d", &t);
	while(t--)
	{
		ll n;
		scanf("%I64d", &n);
		printf("%I64d\n", solve(n+1));
	}
	return 0;
}