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;
}