HDU-3555-數位dp

題目大意：問區間[0,n]中有多少個數中數位包括49；

題目解析：定義dp[i][j]表示有多少個符合條件，i表示第幾位，j=0的時候前面的位沒有49並且上一位不是4，j=1的時候前面的位數沒有49但是上一位是4，j=2的時候表示前面的位數已經包括49了；

AC代碼：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
ll dp[20][3],num[20];
ll dfs(ll pos,ll pre,bool limit)
{
	if(pos==-1)	return pre==2;
	if(!limit&&dp[pos][pre]!=-1)	return dp[pos][pre];
	ll u=limit?num[pos]:9;
	ll ans=0;
	for(ll i=0;i<=u;i++)
	{
		if(pre==2)
			ans+=dfs(pos-1,2,limit&&i==u);
		else if(pre==1&&i==9)
			ans+=dfs(pos-1,2,limit&&i==u);
		else if(i==4)
			ans+=dfs(pos-1,1,limit&&i==u);
		else ans+=dfs(pos-1,0,limit&&i==u);
	}
	if(!limit)	return dp[pos][pre]=ans;
	return ans;
}
ll solve(ll n)
{
	ll cnt=0;
	while(n)
	{
		num[cnt++]=n%10;
		n/=10;
	}
	return dfs(cnt-1,0,true);
}
int main()
{
	int cas;
	scanf("%d",&cas);
	memset(dp,-1,sizeof(dp));
	while(cas--)
	{
		ll n;
		scanf("%lld",&n);
		printf("%lld\n",solve(n));			
	}
	return 0;
}